diff --git a/source/Layer.hpp b/source/Layer.hpp index 76cfbee..ecfcf80 100644 --- a/source/Layer.hpp +++ b/source/Layer.hpp @@ -49,6 +49,11 @@ public: } } + void train(arma::mat const &batch, IListener *pListener=nullptr) + { + Rbm::train(trainingData(batch), pListener); + } + std::string& name() { return m_name;