Skip to content

Commit 57a4834

Browse files
committed
Add Poisson and Gamma deviance to the first CV
1 parent 83ed732 commit 57a4834

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

content/python_files/feature_engineering.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -569,14 +569,19 @@ def build_features(
569569

570570
# %%
571571
from sklearn.metrics import make_scorer, mean_absolute_percentage_error, get_scorer
572+
from sklearn.metrics import d2_tweedie_score
572573

573574

574575
mape_scorer = make_scorer(mean_absolute_percentage_error)
576+
poisson_scorer = make_scorer(d2_tweedie_score, power=1.0)
577+
gamma_scorer = make_scorer(d2_tweedie_score, power=2.0)
575578

576579
cv_results = predictions.skb.cross_validate(
577580
cv=ts_cv_5,
578581
scoring={
579582
"r2": get_scorer("r2"),
583+
"d2_poisson": poisson_scorer,
584+
"d2_gamma": gamma_scorer,
580585
"mape": mape_scorer,
581586
},
582587
return_train_score=True,

0 commit comments

Comments
 (0)