stanfordmlgroup / ngboost

Natural Gradient Boosting for Probabilistic Prediction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Partial fit

bakirillov opened this issue · comments

Hello!

Is partial fit supported? I have a dataset too large to fit into memory, so I would like to fit the model on mini-batches of it.
There is no partial_fit method, would just running fit on every batch work?

this functionality doesn't exist at the moment but it would be pretty easy to change the code to allow it. The fit method is here. Would very much welcome a PR!