tree model updated

MegaJoctan · Jul 31, 2024 · ff83426 · ff83426
1 parent a6c7f25
commit ff83426
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/Decision Tree/tree.mqh b/Decision Tree/tree.mqh
@@ -106,8 +106,9 @@ public:
 
                      void fit(const matrix &x, const vector &y);
                      void print_tree(Node *tree, string indent=" ",string padl="");
-                     double predict_bin(const vector &x);
-                     vector predict_bin(const matrix &x);
+
+                     virtual double predict_bin(const vector &x);
+                     virtual vector predict_bin(const matrix &x);
   };
 //+------------------------------------------------------------------+
 //|                                                                  |
@@ -455,6 +456,10 @@ public:
                     ~CDecisionTreeRegressor(void);
 
                      void fit(matrix &x, vector &y);
+
+                     virtual double predict(const vector &x); //this is used for live trading
+                     virtual vector predict(const matrix &x);
+
   };
 //+------------------------------------------------------------------+
 //|                                                                  |
@@ -597,3 +602,17 @@ double CDecisionTreeRegressor::calculate_leaf_value(vector &Y)
 //+------------------------------------------------------------------+
 //|                                                                  |
 //+------------------------------------------------------------------+
+vector CDecisionTreeRegressor::predict(const matrix &x)
+ {
+  return predict_bin(x);
+ }
+//+------------------------------------------------------------------+
+//|                                                                  |
+//+------------------------------------------------------------------+
+double CDecisionTreeRegressor::predict(const vector &x)
+ {
+  return predict_bin(x);
+ }
+//+------------------------------------------------------------------+
+//|                                                                  |
+//+------------------------------------------------------------------+