- trainingdata always contains context
- load / store training batch with context - on load: add context part to legacy training batches - removed Rbm::setBatch() git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@790 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+1
-2
@@ -116,10 +116,9 @@ void Rbm::contrastiveDivergence(arma::mat const &v_states, arma::mat &dw, arma::
|
||||
dbhv -= sum(h_probs, 0);
|
||||
}
|
||||
|
||||
void Rbm::train(IListener* pListener)
|
||||
void Rbm::train(arma::mat const &batch, IListener* pListener)
|
||||
{
|
||||
Status status;
|
||||
arma::mat const &batch = m_batch;
|
||||
double dProgress = 100.0/(batch.n_rows*m_params.numEpochs);
|
||||
double progress = 0;
|
||||
int lastProgress = -100;
|
||||
|
||||
Reference in New Issue
Block a user