set model in models dict

This commit is contained in:
robcaulk 2022-09-18 12:49:08 +02:00
parent fa3d4b58ab
commit 188f75d8ec

View File

@ -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]