Fix predict

aimclub · May 22, 2024 · 788deb5 · 788deb5
1 parent be78a70
commit 788deb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fedot_ind/core/models/ts_forecasting/markov_ar.py b/fedot_ind/core/models/ts_forecasting/markov_ar.py
@@ -63,11 +63,11 @@ def predict(self, input_data):
         self.handle_new_data(input_data)
         start_id = self.actual_ts_len
         end_id = start_id + forecast_length - 1
-        predicted = self.autoreg.predict(start=start_id, end=end_id)
+        # predicted = self.autoreg.predict(start=start_id, end=end_id)
         # predicted = []
         # for t in range(start_id, end_id):
         #     self.autoreg = self.autoreg(input_data[start_id:t])
-        #     predicted = MSARExtension(self.autoreg).predict_out_of_sample()
+        predicted = MSARExtension(self.autoreg).predict_out_of_sample()
 
         predict = self.scaler.inverse_transform(np.array([predicted]).ravel().reshape(1, -1))
         # predict = np.array(predicted).reshape(1, -1)