Scikits-Learn SVM SVC простой вопрос

У меня есть проблема, которую я решил, но я хочу знать, прав ли я.

В учебной документации Scikit, касающейся SVM SVC, приведен пример управления несбалансированными данными с использованием весов в классах.

они помещают пример, где вес классов сообщается в svm.SVC()

    wclf = svm.SVC(kernel='linear', class_weight={1: 10})

но если воспроизвести эту команду в исходном коде, я получаю следующую ошибку:

    wclf = svm.SVC(kernel='linear', class_weight={1: 10})
    TypeError: __init__() got an unexpected keyword argument 'class_weight'

Но если поместить функцию classes_weight в функцию fit(), проблема решается:

    wclf.fit(X, y, class_weight={1: 10})

я прав по этому поводу? кто-нибудь когда-нибудь имел эту проблему?

1 ответ

Решение

Ключевое слово 'class_weight' еще не реализовано в вашей версии sklearn для SVC, но для SVC.fit(). sklearn обновляет свои функции иногда медленнее, чем вы думаете, и документация, которую вы читаете, может быть /dev/ или /stable/ вместо вашей версии.

Другие вопросы по тегам