adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

throw user error if user tries to load models but feeds the wrong features (while using PCA)

Browse Source
This commit is contained in:
robcaulk
2022-05-28 11:11:41 +02:00
parent b8f9c3557b
commit c5a16e91fb
4 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
freqtrade/freqai/prediction_models/CatboostPredictionModel.py
View File

@@ -71,7 +71,8 @@ class CatboostPredictionModel(IFreqaiModel):
# optional additional data cleaning/analysis
self.data_cleaning_train(dh)
logger.info(f'Training model on {len(dh.training_features_list)} features')
logger.info(f'Training model on {len(dh.data_dictionary["train_features"].columns)}'
'features')
logger.info(f'Training model on {len(data_dictionary["train_features"])} data points')
model = self.fit(data_dictionary)
@@ -129,7 +130,7 @@ class CatboostPredictionModel(IFreqaiModel):
dh.data_dictionary["prediction_features"] = filtered_dataframe
# optional additional data cleaning/analysis
self.data_cleaning_predict(dh)
self.data_cleaning_predict(dh, filtered_dataframe)
predictions = self.model.predict(dh.data_dictionary["prediction_features"])