search

maskot1977 / scikitallstars

MIT License
5 stars 2 forks source link

KeyError: 'kNN' #2

Open cidrugHug8 opened 3 years ago

cidrugHug8 commented 3 years ago

When I let it solve regression task, it gave me the following error.

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
/tmp/ipykernel_83730/4028457873.py in <module>
      6 
      7 from scikitallstars import allstars, depict
----> 8 allstars_model = allstars.fit(X_train, y_train, timeout=1000, n_trials=100, feature_selection=True)

/usr/local/lib/python3.8/dist-packages/scikitallstars-0.1.0-py3.8.egg/scikitallstars/allstars.py in fit(X_train, y_train, feature_selection, verbose, timeout, n_trials, show_progress_bar)
    929             print(
    930                 objective.best_scores[model_name],
--> 931                 objective.best_models[model_name].model,
    932             )
    933 

KeyError: 'kNN'