Skip to content

Latest commit

 

History

History
28 lines (18 loc) · 725 Bytes

README.md

File metadata and controls

28 lines (18 loc) · 725 Bytes

Hurdle

tests status

This package contains an implementation of Hurdle Regression, based in part on Geoff Ruddocks implementation and Håvard Hegres 2022 adaption of his implementation.

Installation

pip install hurdle

To use the hurdle.menu_hurdle_estimator.MenuHurdleEstimator module you also need to install the xgboost and lightgbm packages.

Usage

from hurdle import HurdleEstimator
from sklearn import linear_model

est = HurdleEstimator(linear_model.LogisticRegression(), linear_model.LinearModel())

est.fit(...)