ext/tomoto/ct.cpp in tomoto-0.2.2 vs ext/tomoto/ct.cpp in tomoto-0.2.3
- old
+ new
@@ -1,58 +1,58 @@
#include <CT.h>
-#include <rice/Module.hpp>
+#include <rice/rice.hpp>
#include "utils.h"
void init_ct(Rice::Module& m) {
Rice::define_class_under<tomoto::ICTModel, tomoto::ILDAModel>(m, "CT")
- .define_singleton_method(
+ .define_singleton_function(
"_new",
- *[](size_t tw, size_t k, tomoto::Float alpha, tomoto::Float eta, size_t seed) {
+ [](size_t tw, size_t k, tomoto::Float alpha, tomoto::Float eta, size_t seed) {
tomoto::CTArgs args;
args.k = k;
args.alpha = {alpha};
args.eta = eta;
if (seed >= 0) {
args.seed = seed;
}
return tomoto::ICTModel::create((tomoto::TermWeight)tw, args);
- })
+ }, Rice::Return().takeOwnership())
.define_method(
"_correlations",
- *[](tomoto::ICTModel& self, tomoto::Tid topic_id) {
+ [](tomoto::ICTModel& self, tomoto::Tid topic_id) {
return self.getCorrelationTopic(topic_id);
})
.define_method(
"num_beta_sample",
- *[](tomoto::ICTModel& self) {
+ [](tomoto::ICTModel& self) {
return self.getNumBetaSample();
})
.define_method(
"num_beta_sample=",
- *[](tomoto::ICTModel& self, size_t value) {
+ [](tomoto::ICTModel& self, size_t value) {
self.setNumBetaSample(value);
return value;
})
.define_method(
"num_tmn_sample",
- *[](tomoto::ICTModel& self) {
+ [](tomoto::ICTModel& self) {
return self.getNumTMNSample();
})
.define_method(
"num_tmn_sample=",
- *[](tomoto::ICTModel& self, size_t value) {
+ [](tomoto::ICTModel& self, size_t value) {
self.setNumTMNSample(value);
return value;
})
.define_method(
"_prior_cov",
- *[](tomoto::ICTModel& self) {
+ [](tomoto::ICTModel& self) {
return self.getPriorCov();
})
.define_method(
"prior_mean",
- *[](tomoto::ICTModel& self) {
+ [](tomoto::ICTModel& self) {
return self.getPriorMean();
});
}