Sha256: 03b91d369972446bb5215920919a5aabbec4bed5a57c0d01dde25c0bf66e37ad
Contents?: true
Size: 896 Bytes
Versions: 1
Compression:
Stored size: 896 Bytes
Contents
#include <PLDA.h> #include <rice/Module.hpp> #include "utils.h" void init_plda(Rice::Module& m) { Rice::define_class_under<tomoto::IPLDAModel, tomoto::ILLDAModel>(m, "PLDA") .define_singleton_method( "_new", *[](size_t tw, size_t latent_topics, tomoto::Float alpha, tomoto::Float eta, int seed) { if (seed < 0) { seed = std::random_device{}(); } return tomoto::IPLDAModel::create((tomoto::TermWeight)tw, latent_topics, 1, alpha, eta, seed); }) .define_method( "_add_doc", *[](tomoto::IPLDAModel& self, std::vector<std::string> words, std::vector<std::string> labels) { auto doc = buildDoc(words); doc.misc["labels"] = labels; return self.addDoc(doc); }) .define_method( "latent_topics", *[](tomoto::IPLDAModel& self) { return self.getNumLatentTopics(); }); }
Version data entries
1 entries across 1 versions & 1 rubygems
Version | Path |
---|---|
tomoto-0.1.4 | ext/tomoto/plda.cpp |