- assign training data with context

- calculate context on demand if empty git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@773 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-10 12:19:48 +00:00
parent 9f73824bc4
commit 8c8e0b324d
5 changed files with 21 additions and 12 deletions
@@ -124,6 +124,15 @@ private:
 		patterSlider->setRange(0, m_stack->numTraining()-1, 1);
 	}

+	const arma::mat trainingAt(size_t index)
+	{
+		if (m_pLayer->context().is_empty())
+		{
+			m_pLayer->setBatch(m_stack->trainingData());
+		}
+		return arma::join_rows(m_stack->trainingData().row(index), m_pLayer->context().row(index));
+	}
+	
 	void updateControls();
 	bool onProgress(Rbm *pRbm, const Rbm::Status &status) override;