stanfordmlgroup/ngboost v0.5.11

Bug Fixes in APIs

on GitHub

Version 0.5.11

• Fix NGBClassifier and NGBSurvival API parity by adding validation_fraction and early_stopping_rounds, with regression coverage for their validation-split paths (issue #402)
• Replace the deprecated friedman_mse criterion with squared_error for the default tree learner and distribution tests (issue #408, PR #409)
• Support a distinct base learner for each distribution parameter, including scikit-learn nested parameters and feature importances for all-tree configurations (issue #338, PR #411)