- use expectations
- improved gibbs sampling - LayerArray is template class git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@18 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+12
-4
@@ -38,7 +38,8 @@
|
||||
//[/Comments]
|
||||
*/
|
||||
class MainComponent : public Component,
|
||||
public VisibleLayerArrayListener,
|
||||
public LayerArrayListener<VisibleLayer>,
|
||||
public RbmListener,
|
||||
public ButtonListener,
|
||||
public SliderListener,
|
||||
public LabelListener
|
||||
@@ -67,20 +68,24 @@ private:
|
||||
ScopedPointer<DrawComponent> Draw;
|
||||
ScopedPointer<DrawComponent> Draw2;
|
||||
ScopedPointer<DrawComponent> DrawWeights;
|
||||
Array<VisibleLayer>m_trainingData;
|
||||
uint32_t m_vNumX;
|
||||
uint32_t m_vNumY;
|
||||
uint32_t m_hNum;
|
||||
uint32_t m_vNumX_next;
|
||||
uint32_t m_vNumY_next;
|
||||
uint32_t m_hNum_next;
|
||||
VisibleLayerArray m_layers;
|
||||
LayerArray<VisibleLayer> m_layers;
|
||||
void load();
|
||||
void save();
|
||||
void create();
|
||||
void destroy();
|
||||
const juce::String& getBaseDir();
|
||||
void onChanged(const VisibleLayerArray &obj);
|
||||
void onChanged(const LayerArray<VisibleLayer> &obj);
|
||||
void onEpochTrained(const Rbm &obj);
|
||||
String m_baseDir;
|
||||
double m_trainingProgress;
|
||||
uint32_t m_numGibbs;
|
||||
bool m_rbmUseExpectations;
|
||||
//[/UserVariables]
|
||||
|
||||
//==============================================================================
|
||||
@@ -104,6 +109,9 @@ private:
|
||||
ScopedPointer<TextButton> saveTrainingButton;
|
||||
ScopedPointer<TextButton> clearTrainingButton;
|
||||
ScopedPointer<TextButton> removeTrainingButton;
|
||||
ScopedPointer<Slider> numGibbsSlider;
|
||||
ScopedPointer<TextButton> reconstructEquButton;
|
||||
ScopedPointer<ToggleButton> rbmUseExpectationsToggleButton;
|
||||
|
||||
|
||||
//==============================================================================
|
||||
|
||||
Reference in New Issue
Block a user