- cleaned up

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@625 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-11-07 16:35:00 +00:00
parent 4e21dd8114
commit 6ccffc119d
4 changed files with 13 additions and 105 deletions
+2 -25
View File
@@ -26,7 +26,6 @@
RbmComponent::RbmComponent (const std::string &name, size_t id, size_t numVisibleX, size_t numVisibleY, size_t numHidden, RbmComponentListener *pListener)
: Layer(name, id, numVisibleX, numVisibleY, numHidden)
, m_currWeightIndexToDraw(0)
, m_currTrainingIndexToDraw(0)
, DrawTraining(nullptr)
, DrawReconstruction(nullptr)
, DrawWeights(nullptr)
@@ -249,24 +248,12 @@ void RbmComponent::redrawWeights()
DrawWeights->DrawData();
}
void RbmComponent::onParamsChanged()
{
redrawWeights();
redrawReconstruction();
}
void RbmComponent::setWeightsIndex(size_t index)
void RbmComponent::redrawWeights(size_t index)
{
m_currWeightIndexToDraw = index;
redrawWeights();
}
size_t RbmComponent::getWeightsIndex()
{
return m_currWeightIndexToDraw;
}
void RbmComponent::setTrainingIndex(arma::mat const& batch)
void RbmComponent::setTrainingData(arma::mat const& batch)
{
DrawTraining->getData() = batch;
DrawTraining->DrawData();
@@ -275,16 +262,6 @@ void RbmComponent::setTrainingIndex(arma::mat const& batch)
upPass(DrawHidden->getData());
}
size_t RbmComponent::getTrainingIndex()
{
return m_currTrainingIndexToDraw;
}
arma::mat const& RbmComponent::getTrainingData()
{
return DrawTraining->getData();
}
arma::mat const& RbmComponent::getTopWeights()
{
if (upper)