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

- cleaned up

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@625 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-11-07 16:35:00 +00:00
parent 4e21dd8114
commit 6ccffc119d
4 changed files with 13 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RbmComponent.hpp
+7 -11
View File
@@ -98,15 +98,17 @@ public:
void mouseDoubleClick (const MouseEvent& e) override;
void mouseWheelMove (const MouseEvent& e, const MouseWheelDetails& wheel) override;
void setWeightsIndex(size_t index);
size_t getWeightsIndex();
void train(const arma::mat& batch)
{
Layer::train(batch, this);
}
void setTrainingIndex(arma::mat const& batch);
size_t getTrainingIndex();
void setTrainingData(arma::mat const& batch);
void redrawWeights();
void redrawWeights(size_t index);
void redrawReconstruction();
arma::mat const& getTrainingData();
void downPass(arma::mat &dst, arma::mat const &src) override
{
@@ -133,10 +135,6 @@ public:
arma::mat getConvolutedWeight(arma::mat const &h) override;
arma::mat const& getWeights() override;
void train(const arma::mat& batch)
{
Layer::train(batch, this);
}
private:
//[UserVariables] -- You can add your own custom variables in this section.
RbmComponentListener *m_listener;
@@ -145,8 +143,6 @@ private:
ScopedPointer<DrawComponent> DrawWeights;
ScopedPointer<DrawComponent> DrawHidden;
size_t m_currWeightIndexToDraw;
size_t m_currTrainingIndexToDraw;
void onParamsChanged();
bool onProgress(const Rbm::Status &status) override;
void onDraw(DrawComponent &obj) override;
//[/UserVariables]