У меня есть образец 10000 для одного класса, тогда как только 3500 образцов другого класса. Вот почему я столкнулся с проблемой дисбаланса класса. Есть ли какая-то оштрафованная версия SVM в библиотеке изучения sci-kit?Как наказывать SVM за одну категорию в библиотеке обучения sci-kit?
0
A
ответ
1
В SVC, ключевое слово class-weight
в методе fit
устанавливает параметр C
к C*value
, где value>0
.
На самом деле, class weight
- это словарь в формате class_label: value
.
См. http://scikit-learn.org/stable/modules/svm.html (sec: 1.4.1.3)