Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Scikit-Learn Classification and Regression with Weights

How can I do classification or regression in sklearn if I want to weight each sample differently? Is there a way to do it with a custom loss function? If so, what does that loss function look like in general? Is there an easier way?

like image 530
Thomas Johnson Avatar asked Oct 03 '22 18:10

Thomas Johnson


1 Answers

To weigh individual samples, feed a sample_weight array to the estimator's fit method. This should be a 1-d array of length n_samples (i.e. the same dimension as y in most tasks):

estimator.fit(X, y, sample_weight=some_array)

Not all models support this, check the documentation.

like image 80
Fred Foo Avatar answered Oct 09 '22 11:10

Fred Foo