JuliaAI / MLJScikitLearnInterface.jl

MLJ Interface for ScikitLearn.jl
Other
12 stars 6 forks source link

Add sample weights support for sk-learn models #17

Open ablaom opened 4 years ago

ablaom commented 4 years ago

Here is a list of sk-learn models that can take sample weights in training: https://github.com/alan-turing-institute/MLJ.jl/issues/327

The MLJ API supports sample weights; the convention is to overload fit(model, verbosity, X, y, w) where w is a vector of weights (with same length as y). Search for sample weights in the docs