- refactored

- create training one th e fly from file name

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@860 b431acfa-c32f-4a4a-93f1-934dc6c82436

source/RnnStack.cpp

@@ -84,3 +84,51 @@ arma::mat RnnStack::step_forward(arma::mat &state, const arma::mat& v_curr)
 	}
 	return r;
 }
+
+void RnnStack::clamp_one_hot(arma::mat& srcDst)
+{
+	int index = srcDst.index_max();
+	srcDst = arma::zeros(arma::size(srcDst));
+	srcDst[index] = 1;
+}
+
+void RnnStack::sample_one_hot(arma::mat& srcDst)
+{
+	double k = arma::accu(srcDst);
+
+	if (k != 0)
+	{
+		srcDst = srcDst / k;
+	}
+
+	arma::mat ps = Matutils::sample(srcDst);
+	if (arma::accu(ps) > 1)
+	{
+		arma::uvec q1 = find(ps > 0);
+		int winner = (q1.n_elem - 1) * arma::randu(1)[0];
+		int wix = q1(winner);
+
+		srcDst = zeros(arma::size(srcDst));
+		srcDst[wix] = 1;
+	}
+}
+
+arma::mat RnnStack::to_curr(const arma::mat& v)
+{
+	return v.cols(NUM_CODES, 2 * NUM_CODES - 1);
+}
+
+arma::mat RnnStack::to_next(const arma::mat& v)
+{
+	return v.cols(0, NUM_CODES - 1);
+}
+
+void RnnStack::setParams(const Rbm::Params& param)
+{
+	Layer *pLayer = getLayer(0);
+	while (pLayer)
+	{
+		pLayer->params() = param;
+		pLayer = pLayer->next;
+	}
+}