Sha256: a4bd82816d900e6d3b665898554a772f068f437c8fab46efc6eae67feb62d546
Contents?: true
Size: 1.2 KB
Versions: 25
Compression:
Stored size: 1.2 KB
Contents
#include <HDP.h>
#include <rice/rice.hpp>
#include "utils.h"
void init_hdp(Rice::Module& m) {
Rice::define_class_under<tomoto::IHDPModel, tomoto::ILDAModel>(m, "HDP")
.define_singleton_function(
"_new",
[](size_t tw, size_t k, tomoto::Float alpha, tomoto::Float eta, tomoto::Float gamma, size_t seed) {
tomoto::HDPArgs args;
args.k = k;
args.alpha = {alpha};
args.eta = eta;
args.gamma = gamma;
if (seed >= 0) {
args.seed = seed;
}
return tomoto::IHDPModel::create((tomoto::TermWeight)tw, args);
}, Rice::Return().takeOwnership())
.define_method(
"alpha",
[](tomoto::IHDPModel& self) {
return self.getAlpha();
})
.define_method(
"gamma",
[](tomoto::IHDPModel& self) {
return self.getGamma();
})
.define_method(
"live_k",
[](tomoto::IHDPModel& self) {
return self.getLiveK();
})
.define_method(
"live_topic?",
[](tomoto::IHDPModel& self, size_t tid) {
return self.isLiveTopic(tid);
})
.define_method(
"num_tables",
[](tomoto::IHDPModel& self) {
return self.getTotalTables();
});
}
Version data entries
25 entries across 25 versions & 1 rubygems