SKLearn: TypeError: __init__() got an unexpected keyword argument n_splits

Ich versuche, SKLearn (version 0.18.1) wie folgt:

from sklearn.model_selection import KFold
kfold = KFold(n_splits=5, random_state=100)

Aber bekomme ich diese seltsame Fehlermeldung:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-17-b8dd4f5596be> in <module>()
----> 1 kfold = KFold(k=5, random_state=100)
      2 results = cross_val_score(estimator, X, Y, cv=kfold)
      3 print("Results: %.2f (%.2f) MSE" % (results.mean(), results.std()))

TypeError: __init__() got an unexpected keyword argument 'k'

Habe ich konsultiert, die docs hier:

http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.KFold.html

sowie n_splits sieht aus wie ein parameter I sollte in der Lage sein, um pass...

Irgendeine Idee, was hier Los ist /wie beheben?

Dank!

  • was ist Ihre version von sklearn? Sie vielleicht auf der Suche in den docs für eine neuere version als Ihre installierte, könnte man nur ändern, um kfold = KFold(k=5, random_state=100) die sollte funktionieren
InformationsquelleAutor bclayman | 2017-07-12
Schreibe einen Kommentar