interpretml / FasterRisk

Create sparse and accurate risk scoring systems!
BSD 3-Clause "New" or "Revised" License
31 stars 4 forks source link

lb and ub as lists, add wrapper.py and score_visuals.py for more functionality #4

Closed MuhangTian closed 1 year ago

MuhangTian commented 1 year ago

Changes

Allow lb and ub to be passed as lists in RiskScoreOptimizer and starRaySearchModel. This enables user to define their own constraints and use monotonicity constraints.

New Updates

score_visuals.py supports visualization of score card, and FasterRisk class in wrapper.py puts all those things together for FasterRisk (training, printing score, visualizing, getting models...etc) . See more detail in added_functionalities_example.ipynb

Requirements

for visualization, need those packages for support:

More detail is in requirements.txt for reference.