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

Retrain model if FreqAI found a pretrained model but user strategy is not passing the expected features (user has changed the features in the strategy but has passed a the same config[freqai][identifier]). Logger warning output to user.

Browse Source
This commit is contained in:
robcaulk 2022-05-19 17:15:50 +02:00
parent c708dd3186
commit 89eacf2f47
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
freqtrade/freqai/freqai_interface.py
View File

@ -92,6 +92,11 @@ class IFreqaiModel(ABC):
self.dh.save_data(self.model) self.dh.save_data(self.model)
else: else:
self.model = self.dh.load_data() self.model = self.dh.load_data()
strategy_provided_features = self.dh.find_features(dataframe_train)
if strategy_provided_features != self.dh.training_features_list:
logger.info("User changed input features, retraining model.")
self.model = self.train(dataframe_train, metadata)
self.dh.save_data(self.model)
preds, do_preds = self.predict(dataframe_backtest, metadata) preds, do_preds = self.predict(dataframe_backtest, metadata)