lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

ensure predict_proba follows suit. Remove all lib specific params from example config

Browse Source
This commit is contained in:
robcaulk 2022-12-11 12:39:31 +01:00
parent 85f22b5c30
commit 8c7ec07951
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
config_examples/config_freqai.example.json
View File

@ -80,7 +80,6 @@
"random_state": 1
},
"model_training_parameters": {
"n_estimators": 1000
}
},
"bot_name": "",

2
freqtrade/freqai/base_models/BaseClassifierModel.py
View File

@ -101,6 +101,8 @@ class BaseClassifierModel(IFreqaiModel):
pred_df = DataFrame(predictions, columns=dk.label_list)
predictions_prob = self.model.predict_proba(dk.data_dictionary["prediction_features"])
if self.CONV_WIDTH == 1:
predictions_prob = np.reshape(predictions_prob, (-1, len(self.model.classes_)))
pred_df_prob = DataFrame(predictions_prob, columns=self.model.classes_)
pred_df = pd.concat([pred_df, pred_df_prob], axis=1)