CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-02-11 08:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #9704 from ShaneOss/patch-2

Browse Source 
fix: logical error in the conditional checks for model classes
This commit is contained in:
Matthias
2024-02-06 19:53:04 +01:00
committed by GitHub
parent d2e9d36dea cb2aaa7bbb
commit 9e89567eeb
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
freqtrade/freqai/utils.py
View File

@@ -118,10 +118,12 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
mdl = models[label]
if "catboost.core" in str(mdl.__class__):
feature_importance = mdl.get_feature_importance()
elif "lightgbm.sklearn" or "xgb" in str(mdl.__class__):
elif "lightgbm.sklearn" in str(mdl.__class__):
feature_importance = mdl.feature_importances_
elif "xgb" in str(mdl.__class__):
feature_importance = mdl.feature_importances_
else:
logger.info('Model type not support for generating feature importances.')
logger.info('Model type does not support generating feature importances.')
return
# Data preparation