vendor/tomotopy/src/TopicModel/PLDA.h in tomoto-0.1.4 vs vendor/tomotopy/src/TopicModel/PLDA.h in tomoto-0.2.0
- old
+ new
@@ -1,16 +1,27 @@
#pragma once
#include "LLDA.h"
namespace tomoto
{
+ struct PLDAArgs : public LDAArgs
+ {
+ size_t numLatentTopics = 0;
+ size_t numTopicsPerLabel = 1;
+ PLDAArgs setK(size_t _k = 1) const
+ {
+ PLDAArgs ret = *this;
+ ret.k = _k;
+ return ret;
+ }
+ };
+
class IPLDAModel : public ILLDAModel
{
public:
using DefaultDocType = DocumentLLDA<TermWeight::one>;
- static IPLDAModel* create(TermWeight _weight, size_t _numLatentTopics = 0, size_t _numTopicsPerLabel = 1,
- Float alpha = 0.1, Float eta = 0.01, size_t seed = std::random_device{}(),
+ static IPLDAModel* create(TermWeight _weight, const PLDAArgs& args,
bool scalarRng = false);
virtual size_t getNumLatentTopics() const = 0;
};
}
\ No newline at end of file