2014-10-27 2 views
0

У меня есть следующий код, чтобы узнать из моего набора данных:ValueError: нужно больше, чем 1 значение для распаковки с классификатором в scikit учиться

>>> train_features[:5] 
array([[2.0, 9.0, 37.0, 0.0, 28.71, 0.0, 243.63, False], 
     [2.0, 0.0, 4.0, 0.0, 0.0, 0.0, 6.3100000000000005, False], 
     [2.0, 3.0, 3.0, 0.0, 28.07, 0.0, 28.07, False], 
     [2.0, 1.0, 2.0, 0.0, 5.49, 0.0, 14.48, False], 
     [2.0, 3.0, 3.0, 0.0, 7.4700000000000015, 0.0, 7.4700000000000015, 
     False]], dtype=object) 

>>> train_labels[:5] 
array([ True, False, True, False, True], dtype=bool) 

>>> rf = RandomForestClassifier(n_estimators=10) 
>>> rf.fit(train_labels, train_features) 

я получаю эту ошибку на пригонках функции:

ValueError: need more than 1 value to unpack

Я считаю, что это ошибка форматирования. Какую ценность ожидает scikit-learn? Я не нашел входную ссылку в руководстве scikit-learn.

ответ

3

Ошибка только в том, что вы передали аргументы в обратном порядке. Заменить:

rf.fit(train_labels, train_features) 

по:

rf.fit(train_features,train_labels) 

Надеется, что это устраняет проблему.

Смежные вопросы