jrzaurin / pytorch-widedeep

A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch
Apache License 2.0
1.27k stars 188 forks source link

Add Regression losses #59

Closed 5uperpalo closed 2 years ago

5uperpalo commented 2 years ago

Add losses that show promising results for regression prediction in scenarios with highly imbalanced datasets used e.g. in Life Time Value prediction

5uperpalo commented 2 years ago

additional notes:

tweedie loss

ZILNloss

QuantileLoss

FINALL UPDATE:

5uperpalo commented 2 years ago

additional notes for TweedieLoss:

Summary