search

cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.
http://contextualized.ml/
GNU General Public License v3.0
65 stars 9 forks source link

Custom loss functions (and ideally, task-specific) #55

Open blengerich opened 2 years ago

blengerich commented 2 years ago

Would be great to have customizable loss functions, and if possible, task-specific loss functions.

cnellington commented 2 years ago

Notes from last discussion: easiest implementation would be to split tasks from the dataloader into T_x and T_y and pass these to a generic loss function which handles T_x, T_y, and y_hat to route predictions to task-specific losses.