We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 83ed732 commit 57a4834Copy full SHA for 57a4834
1 file changed
content/python_files/feature_engineering.py
@@ -569,14 +569,19 @@ def build_features(
569
570
# %%
571
from sklearn.metrics import make_scorer, mean_absolute_percentage_error, get_scorer
572
+from sklearn.metrics import d2_tweedie_score
573
574
575
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)
578
579
cv_results = predictions.skb.cross_validate(
580
cv=ts_cv_5,
581
scoring={
582
"r2": get_scorer("r2"),
583
+ "d2_poisson": poisson_scorer,
584
+ "d2_gamma": gamma_scorer,
585
"mape": mape_scorer,
586
},
587
return_train_score=True,
0 commit comments