koaning / scikit-lego

Extra blocks for scikit-learn pipelines.
https://koaning.github.io/scikit-lego/
MIT License
1.28k stars 118 forks source link

patch: `ZeroInflatedRegressor` fix training indices #666

Closed FBruzzesi closed 6 months ago

FBruzzesi commented 6 months ago

Description

Issue reported in #664 is very reasonable. I tried to take a look at when ZeroInflatedRegressor was introduced in #442, but I couldn't find any related discussion on such choice.

Fix #664

Type of change