jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- rename Layer into RbmLayer

Browse Source 
- add some log info on toJson()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@582 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-26 06:07:13 +00:00
parent eed6956971
commit d16f0de8fa
8 changed files with 38 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Stack.cpp
+6 -5
View File
@@ -31,7 +31,7 @@ Stack::~Stack()
{
}
void Stack::addLayer(Layer *pOtherLayer)
void Stack::addLayer(RbmLayer *pOtherLayer)
{
if (!m_pLayers)
{
@@ -40,7 +40,7 @@ void Stack::addLayer(Layer *pOtherLayer)
}
else
{
Layer *pLayer = m_pLayers;
RbmLayer *pLayer = m_pLayers;
while(pLayer->upper)
{
pLayer = pLayer->upper;
@@ -50,9 +50,9 @@ void Stack::addLayer(Layer *pOtherLayer)
}
}
const Layer* Stack::getLayer(size_t id)
const RbmLayer* Stack::getLayer(size_t id)
{
Layer *pLayer = m_pLayers;
RbmLayer *pLayer = m_pLayers;
while(pLayer)
{
if (pLayer->id() == id)
@@ -67,6 +67,7 @@ const Layer* Stack::getLayer(size_t id)
void Stack::save(size_t numTraining)
{
std::cout << "Exporting Project " << m_prjname << std::endl;
ofstream ofs(m_prjname + string(".prj"));
Json::StyledWriter writer;
@@ -76,7 +77,7 @@ void Stack::save(size_t numTraining)
project["stack"]["training_file"] = m_prjname + string(".training.dat");
Json::Value layers(Json::arrayValue);
Layer *pLayer = m_pLayers;
RbmLayer *pLayer = m_pLayers;
while(pLayer)
{
layers.append(pLayer->toJson());