vendor/tomotopy/src/TopicModel/HDP.h in tomoto-0.1.4 vs vendor/tomotopy/src/TopicModel/HDP.h in tomoto-0.2.0
- old
+ new
@@ -54,15 +54,24 @@
}
template<typename _TopicModel> void update(WeightType* ptr, const _TopicModel& mdl);
};
+ struct HDPArgs : public LDAArgs
+ {
+ Float gamma = 0.1;
+
+ HDPArgs()
+ {
+ k = 2;
+ }
+ };
+
class IHDPModel : public ILDAModel
{
public:
using DefaultDocType = DocumentHDP<TermWeight::one>;
- static IHDPModel* create(TermWeight _weight, size_t _K = 1,
- Float alpha = 0.1, Float eta = 0.01, Float gamma = 0.1, size_t seed = std::random_device{}(),
+ static IHDPModel* create(TermWeight _weight, const HDPArgs& args,
bool scalarRng = false);
virtual Float getGamma() const = 0;
virtual size_t getTotalTables() const = 0;
virtual size_t getLiveK() const = 0;
\ No newline at end of file