Scikit-learn cross val score: too many indices for array

前端 未结 5 1869
再見小時候
再見小時候 2020-12-15 05:15

I have the following code

 from sklearn.ensemble import ExtraTreesClassifier
 from sklearn.cross_validation import cross_val_score
 #split the dataset for tr         


        
5条回答
  •  青春惊慌失措
    2020-12-15 05:44

    You might need to play with the dimensions a bit, e.g.

    et_score = cross_val_score(et, features, labels, n_jobs=-1)[:,n]
    

    or

     et_score = cross_val_score(et, features, labels, n_jobs=-1)[n,:]
    

    n being the dimension.

提交回复
热议问题