Sha256: eae4de37a1d58317dba8f829bf48412bc954f61920edbacd6bb1acf10534f838
Contents?: true
Size: 1.43 KB
Versions: 24
Compression:
Stored size: 1.43 KB
Contents
#include <LLDA.h>
#include <rice/rice.hpp>
#include "utils.h"
void init_llda(Rice::Module& m) {
Rice::define_class_under<tomoto::ILLDAModel, tomoto::ILDAModel>(m, "LLDA")
.define_singleton_function(
"_new",
[](size_t tw, size_t k, tomoto::Float alpha, tomoto::Float eta, size_t seed) {
tomoto::LDAArgs args;
args.k = k;
args.alpha = {alpha};
args.eta = eta;
if (seed >= 0) {
args.seed = seed;
}
return tomoto::ILLDAModel::create((tomoto::TermWeight)tw, args);
}, Rice::Return().takeOwnership())
.define_method(
"_add_doc",
[](tomoto::ILLDAModel& 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(
"topics_per_label",
[](tomoto::ILLDAModel& self) {
return self.getNumTopicsPerLabel();
})
.define_method(
"topic_label_dict",
[](tomoto::ILLDAModel& self) {
auto dict = self.getTopicLabelDict();
Array res;
auto utf8 = Rice::Class(rb_cEncoding).call("const_get", "UTF_8");
for (size_t i = 0; i < dict.size(); i++) {
VALUE value = Rice::detail::To_Ruby<std::string>().convert(dict.toWord(i));
Object obj(value);
res.push(obj.call("force_encoding", utf8));
}
return res;
});
}
Version data entries
24 entries across 24 versions & 1 rubygems