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

simplify plot_feature_importance call

Browse Source
This commit is contained in:
initrv
2022-09-17 18:53:43 +03:00
parent 86aa875bc9
commit 1c92734f39
2 changed files with 20 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
freqtrade/freqai/freqai_interface.py
View File

@@ -556,14 +556,6 @@ class IFreqaiModel(ABC):
model = self.train(unfiltered_dataframe, pair, dk)
if self.freqai_info["feature_parameters"].get("plot_feature_importance", False):
plot_feature_importance(
model=model,
feature_names=dk.training_features_list,
pair=pair,
train_dir=dk.data_path
)
self.dd.pair_dict[pair]["trained_timestamp"] = new_trained_timerange.stopts
dk.set_new_model_names(pair, new_trained_timerange)
self.dd.pair_dict[pair]["first"] = False
@@ -571,6 +563,9 @@ class IFreqaiModel(ABC):
self.dd.pair_to_end_of_training_queue(pair)
self.dd.save_data(model, pair, dk)
if self.freqai_info["feature_parameters"].get("plot_feature_importance", False):
plot_feature_importance(model, pair, dk)
if self.freqai_info.get("purge_old_models", False):
self.dd.purge_old_models()