- Added gaussian hidden units

source/Rbm.hpp

@@ -29,6 +29,8 @@ public:
 		: learningRate(0.1)
 		, weightDecay(0.0)
 		, momentum(0.5)
+		, doGaussianHidden(false)
+		, doGaussianVisible(false)
 		, doRaoBlackwell(true)
 		, gibbsDoSampleVisible(false)
 		, gibbsDoSampleHidden(true)
@@ -46,6 +48,8 @@ public:
 			params["weightDecay"] = weightDecay;
 			params["learningRate"] = learningRate;
 			params["momentum"] = momentum;
+			params["doGaussianVisible"] = (int)doGaussianVisible;
+			params["doGaussianHidden"] = (int)doGaussianHidden;
 			params["doRaoBlackwell"] = (int)doRaoBlackwell;
 			params["gibbsDoSampleVisible"] = (int)gibbsDoSampleVisible;
 			params["gibbsDoSampleHidden"] = (int)gibbsDoSampleHidden;
@@ -63,6 +67,8 @@ public:
 			weightDecay = params.get("weightDecay", weightDecay).asDouble();
 			learningRate = params.get("learningRate", learningRate).asDouble();
 			momentum = params.get("momentum", momentum).asDouble();
+			doGaussianVisible = params.get("doGaussianVisible", doGaussianVisible) == 1;
+			doGaussianHidden = params.get("doGaussianHidden", doGaussianHidden) == 1;
 			doRaoBlackwell = params.get("doRaoBlackwell", doRaoBlackwell) == 1;
 			gibbsDoSampleVisible = params.get("gibbsDoSampleVisible", gibbsDoSampleVisible) == 1;
 			gibbsDoSampleHidden = params.get("gibbsDoSampleHidden", gibbsDoSampleHidden) == 1;
@@ -75,6 +81,8 @@ public:
 		double weightDecay;
 		double learningRate;
 		double momentum;
+		bool doGaussianVisible;
+		bool doGaussianHidden;
 		bool doRaoBlackwell;
 		bool gibbsDoSampleVisible;
 		bool gibbsDoSampleHidden;