Merge pull request #174 from jucendrero/patch-1

Fixed an issue that prevents setting user defined values for nuSq
bab2min · Jul 9, 2022 · 3d50ee1 · 3d50ee1
2 parents d30964c + 78d763e
commit 3d50ee1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/TopicModel/SLDAModel.hpp b/src/TopicModel/SLDAModel.hpp
@@ -379,11 +379,11 @@ namespace tomoto
 			{
 				nuSq = Vector::Ones(F);
 			}
-			else if (args.mu.size() == 1)
+			else if (args.nuSq.size() == 1)
 			{
 				nuSq = Vector::Constant(F, args.nuSq[0]);
 			}
-			else if (args.mu.size() == F)
+			else if (args.nuSq.size() == F)
 			{
 				nuSq = Eigen::Map<const Vector>(args.nuSq.data(), args.nuSq.size());
 			}