search

microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

what is it unreduced_pinball vs tfa.losses.PinballLoss #1068

Open Sandy4321 opened 2 years ago

Sandy4321 commented 2 years ago

what is it unreduced_pinball

check_same_as_tensorflow(
    partial(losses.unreduced_pinball, tau=tau), tfa.losses.PinballLoss(tau=tau)
)

from https://github.com/microsoft/knossos-ksc/blob/09b95f9df85305973101b72eb466c532e0b9438f/rlo/test/rlo/test_losses.py

sarahnlewis commented 2 years ago

unreduced_pinball is a pytorch implementation of pinball loss. The function is here. This test is left over from when we ported code from tensorflow to pytorch, and wanted to check we had equivalent implementations of the loss functions.