[Python] Support balanced weights?

Xtra-Computing / thundersvm

ThunderSVM: A Fast SVM Library on GPUs and CPUs

Apache License 2.0

1.56k stars 216 forks source link

Closed dourouc05 closed 5 years ago

dourouc05 commented 5 years ago

scikit-learn supports class_weight=‘balanced’ (https://scikit-learn.org/stable/modules/generated/sklearn.svm.SVC.html), in which case it deals with the computation of the weights. However, in thundersvm, using this option yields an error. (Which is known: #91.)

Would you be interested in a patch to support this?

QinbinLi commented 5 years ago

Hi @dourouc05 ,

We have added the function in the newest library. You can update the code and try it. Thanks.

dourouc05 commented 5 years ago

Thanks for your reactivity :)!

Will you issue a new version (I see that 0.2.0 dates back from 2017)?