Embed estimator and algo types into jsonized state

WojciechMigda · Aug 23, 2020 · bba0b7b · bba0b7b
1 parent 3863f2e
commit bba0b7b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/src/tsetlini_state_json.cpp b/lib/src/tsetlini_state_json.cpp
@@ -283,6 +283,9 @@ std::string to_json_string(ClassifierStateClassic const & state)
 {
     json js;
 
+    js["estimator"] = "classifier";
+    js["algo"] = "classic";
+
     js["ta_state"] = state.ta_state;
     js["igen"] = state.igen;
     js["fgen"] = state.fgen;
@@ -325,6 +328,9 @@ std::string to_json_string(RegressorStateClassic const & state)
 {
     json js;
 
+    js["estimator"] = "regressor";
+    js["algo"] = "classic";
+
     js["ta_state"] = state.ta_state;
     js["igen"] = state.igen;
     js["fgen"] = state.fgen;
@@ -367,6 +373,9 @@ std::string to_json_string(RegressorStateBitwise const & state)
 {
     json js;
 
+    js["estimator"] = "regressor";
+    js["algo"] = "bitwise";
+
     js["ta_state"] = state.ta_state;
     js["igen"] = state.igen;
     js["fgen"] = state.fgen;
@@ -409,6 +418,9 @@ std::string to_json_string(ClassifierStateBitwise const & state)
 {
     json js;
 
+    js["estimator"] = "classifier";
+    js["algo"] = "bitwise";
+
     js["ta_state"] = state.ta_state;
     js["igen"] = state.igen;
     js["fgen"] = state.fgen;