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

ensure continued operation despite not being able to plot

Browse Source
This commit is contained in:
robcaulk 2022-09-18 00:00:14 +02:00
parent 2c23effbf2
commit 68f7a31504
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
freqtrade/freqai/utils.py
View File

@ -156,6 +156,8 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
if 'FreqaiMultiOutputRegressor' in str(model.__class__):
for estimator, label in zip(model.estimators_, dk.label_list):
models[label] = estimator
else:
models[dk.label_list[0]]
for label in models:
mdl = models[label]
@ -164,8 +166,8 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
elif "lightgbm.sklearn" or "xgb" in str(mdl.__class__):
feature_importance = mdl.feature_importances_
else:
# TODO: Add support for more libraries
raise NotImplementedError(f"Cannot extract feature importance from {mdl.__class__}")
logger.info('Model type not support for generating feature importances.')
return
# Data preparation
fi_df = pd.DataFrame({