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 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__): if 'FreqaiMultiOutputRegressor' in str(model.__class__):
for estimator, label in zip(model.estimators_, dk.label_list): for estimator, label in zip(model.estimators_, dk.label_list):
models[label] = estimator models[label] = estimator
else:
models[dk.label_list[0]]
for label in models: for label in models:
mdl = models[label] 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__): elif "lightgbm.sklearn" or "xgb" in str(mdl.__class__):
feature_importance = mdl.feature_importances_ feature_importance = mdl.feature_importances_
else: else:
# TODO: Add support for more libraries logger.info('Model type not support for generating feature importances.')
raise NotImplementedError(f"Cannot extract feature importance from {mdl.__class__}") return
# Data preparation # Data preparation
fi_df = pd.DataFrame({ fi_df = pd.DataFrame({