set model in models dict
This commit is contained in:
parent
fa3d4b58ab
commit
188f75d8ec
@ -157,7 +157,7 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
|
|||||||
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:
|
else:
|
||||||
models[dk.label_list[0]]
|
models[dk.label_list[0]] = model
|
||||||
|
|
||||||
for label in models:
|
for label in models:
|
||||||
mdl = models[label]
|
mdl = models[label]
|
||||||
|
Loading…
Reference in New Issue
Block a user