sklearn-rvm icon indicating copy to clipboard operation
sklearn-rvm copied to clipboard

will it work for unbalanced data for classifier?

Open Sandy4321 opened this issue 5 years ago • 1 comments

will it work for unbalanced data for classifier? for one hot data: only 0s and 1s in data is it possible to set class weights like https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html class_weight dict or ‘balanced’, default=None Weights associated with classes in the form {class_label: weight}. If not given, all classes are supposed to have weight one

Sandy4321 avatar Mar 09 '20 17:03 Sandy4321

It depends on how unbalanced the data is but it should work.

Currently we don't have class_weight implemented but it is definitely something that we will look into adding in the next versions of the model.

PedroFerreiradaCosta avatar Mar 11 '20 17:03 PedroFerreiradaCosta