fixed #136

bab2min · Sep 1, 2021 · 78ab9f7 · 78ab9f7
1 parent 2289df4
commit 78ab9f7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/python/py_PT.cpp b/src/python/py_PT.cpp
@@ -22,6 +22,8 @@ static int PT_init(TopicModelObject *self, PyObject *args, PyObject *kwargs)
 			[=]() { return "`alpha` must be an instance of `float` or `List[float]` with length `k` (given " + py::repr(objAlpha) + ")"; }
 		);
 
+		if (margs.p == 0) margs.p = margs.k * 10;
+
 		tomoto::ITopicModel* inst = tomoto::IPTModel::create((tomoto::TermWeight)tw, margs);
 		if (!inst) throw py::ValueError{ "unknown `tw` value" };
 		self->inst = inst;