Merge pull request #3 from blei-lab/obs-weights-config

fix obs_weights being ignored in config
blei-lab · Jan 30, 2018 · 93d8456 · 93d8456
2 parents 2324a6f + 44cff40
commit 93d8456
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/def_model.cpp b/def_model.cpp
@@ -207,9 +207,9 @@ void DEFModel::init() {
     return GET_PRIOR_LAYER(layer_type, options, initializer);
   };
   // sparse prior distribution for obs_weight
-  w_obs_layer = build_prior_layer("DEF_weights");
-  if (ptree.get_child("DEF_weights").get<string>("type") == "exp") {
-    w_obs_layer = build_prior_layer("DEF_weights");
+  w_obs_layer = build_prior_layer("obs_weights");
+  if (ptree.get_child("obs_weights").get<string>("type") == "exp") {
+    w_obs_layer = build_prior_layer("obs_weights");
   }
   else {
     w_obs_layer_b = NULL;