diff --git a/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py b/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py index c83a19bb5..d3b5633c3 100644 --- a/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py +++ b/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py @@ -3,7 +3,7 @@ from sklearn.base import is_classifier from sklearn.multioutput import MultiOutputClassifier, _fit_estimator from sklearn.utils.multiclass import check_classification_targets from sklearn.utils.parallel import Parallel, delayed -from sklearn.utils.validation import has_fit_parameter +from sklearn.utils.validation import has_fit_parameter, validate_data from freqtrade.exceptions import OperationalException @@ -36,7 +36,7 @@ class FreqaiMultiOutputClassifier(MultiOutputClassifier): if not hasattr(self.estimator, "fit"): raise ValueError("The base estimator should implement a fit method") - y = self._validate_data(X="no_validation", y=y, multi_output=True) + y = validate_data(self, X="no_validation", y=y, multi_output=True) if is_classifier(self): check_classification_targets(y) diff --git a/freqtrade/freqai/base_models/FreqaiMultiOutputRegressor.py b/freqtrade/freqai/base_models/FreqaiMultiOutputRegressor.py index 32ab0e16d..525cee52d 100644 --- a/freqtrade/freqai/base_models/FreqaiMultiOutputRegressor.py +++ b/freqtrade/freqai/base_models/FreqaiMultiOutputRegressor.py @@ -1,6 +1,6 @@ from sklearn.multioutput import MultiOutputRegressor, _fit_estimator from sklearn.utils.parallel import Parallel, delayed -from sklearn.utils.validation import has_fit_parameter +from sklearn.utils.validation import has_fit_parameter, validate_data class FreqaiMultiOutputRegressor(MultiOutputRegressor): @@ -31,7 +31,7 @@ class FreqaiMultiOutputRegressor(MultiOutputRegressor): if not hasattr(self.estimator, "fit"): raise ValueError("The base estimator should implement a fit method") - y = self._validate_data(X="no_validation", y=y, multi_output=True) + y = validate_data(self, X="no_validation", y=y, multi_output=True) if y.ndim == 1: raise ValueError(