ext/quarry/quarry_toolkit.cpp in thera-0.0.2 vs ext/quarry/quarry_toolkit.cpp in thera-0.0.3

@@ -58,11 +58,11 @@
     
     // text pipeline
     rb_mText.define_module_function("standard_pipeline", &Preprocessing::Text::StandardPipeline);
     Data_Type<Preprocessing::Text::TextPipeline> rb_cTextPipeline = define_class_under<Preprocessing::Text::TextPipeline>(rb_mQuarry, "ImplTextPipeline")
       .define_constructor(Constructor<Preprocessing::Text::TextPipeline>());
-//      .define_method("process_text", &Preprocessing::Text::TextPipeline::process_text);
+      //.define_method("process_text", &Preprocessing::Text::TextPipeline::process_text);
     
     // storage
     Data_Type<Storage::Storage> rb_cStorage = define_class_under<Storage::Storage>(rb_mQuarry, "ImplStorage");
     
     Data_Type<Storage::ARFF> rb_cARFF = define_class_under<Storage::ARFF, Storage::Storage>(rb_mQuarry, "ImplARFF")
@@ -88,10 +88,12 @@
       .define_constructor(Constructor<Model::Model>())
       .define_method("train", &Model::Model::train)
       .define_method("train_text", &Model::Model::train_text)
       .define_method("classify", &Model::Model::classify)
       .define_method("classify_text", &Model::Model::classify_text)
+      .define_method("process_text", &Model::Model::process_text)
+      .define_method("add_text_example", &Model::Model::add_text_example)
       .define_method("set_data_set", &Model::Model::set_data_set)
       .define_method("get_data_set", &Model::Model::get_data_set)
       .define_method("set_classifier", &Model::Model::set_classifier)
       .define_method("get_classifier", &Model::Model::get_classifier)
       .define_method("set_text_pipeline", &Model::Model::set_text_pipeline)
@@ -108,9 +110,10 @@
       .define_method("set_name", &DataSet::Feature::set_name)
       .define_constructor(Constructor<DataSet::Feature, string, int>());
     
     Data_Type<DataSet::Example> rb_cDataSetExample = define_class_under<DataSet::Example>(rb_mDataSet, "ImplExample")
       .define_method("category_index", &DataSet::Example::category_index)
+      .define_method("set_category_index", &DataSet::Example::set_category_index)
       .define_method("get_value", &DataSet::Example::get_value)
       .define_method("set_value", &DataSet::Example::set_value)
       .define_constructor(Constructor<DataSet::Example, int>());
     
     Data_Type<DataSet::DataSet> rb_cDataSet = define_class_under<DataSet::DataSet>(rb_mDataSet, "ImplDataSet")