diff --git a/source/Rbm.hpp b/source/Rbm.hpp index 344da56..a08e49b 100644 --- a/source/Rbm.hpp +++ b/source/Rbm.hpp @@ -61,16 +61,16 @@ public: void fromJson(Json::Value params) { std::cout << "Importing Rbm::Params" << std::endl; - weightDecay = params["weightDecay"].asDouble(); - learningRate = params["learningRate"].asDouble(); - momentum = params["momentum"].asDouble(); - doRaoBlackwell = params["doRaoBlackwell"] == 1; - gibbsDoSampleVisible = params["gibbsDoSampleVisible"] == 1; - gibbsDoSampleHidden = params["gibbsDoSampleHidden"] == 1; - doSampleBatch = params["doSampleBatch"] == 1; - numGibbs = params["numGibbs"].asUInt(); - miniBatchSize = params["miniBatchSize"].asUInt(); - numEpochs = params["numEpochs"].asUInt(); + weightDecay = params.get("weightDecay", weightDecay).asDouble(); + learningRate = params.get("learningRate", learningRate).asDouble(); + momentum = params.get("momentum", momentum).asDouble(); + doRaoBlackwell = params.get("doRaoBlackwell", doRaoBlackwell) == 1; + gibbsDoSampleVisible = params.get("gibbsDoSampleVisible", gibbsDoSampleVisible) == 1; + gibbsDoSampleHidden = params.get("gibbsDoSampleHidden", gibbsDoSampleHidden) == 1; + doSampleBatch = params.get("doSampleBatch", doSampleBatch) == 1; + numGibbs = params.get("numGibbs", (int)numGibbs).asUInt(); + miniBatchSize = params.get("miniBatchSize", (int)miniBatchSize).asUInt(); + numEpochs = params.get("numEpochs", (int)numEpochs).asUInt(); } double weightDecay;