Incorporate Sample Weights

parrt / dtreeviz

A python library for decision tree visualization and model interpretation.

MIT License

2.89k stars 333 forks source link

Open mepland opened 1 year ago

mepland commented 1 year ago

We should allow users to provide a sample_weight vector of the same shape as y_train to incorporate weighted samples in dtreeviz plots.