def quantile_loss(self, a, b): # a가 target, b가 prediction
quantiles_used = set(self.quantiles)
loss = 0.
for i, quantile in enumerate(valid_quantiles):
if quantile in quantiles_used:
loss += utils.tensorflow_quantile_loss(
a[Ellipsis,
output_size * i:output_size * (i + 1)],
b[Ellipsis,
output_size * i:output_size * (i + 1)],
quantile)
return loss
def quantile_loss(self, a, b):
"""Returns quantile loss for specified quantiles.
Args:
a: Targets
b: Predictions
"""
quantiles_used = set(self.quantiles)
loss = 0.
for i, quantile in enumerate(valid_quantiles):
if quantile in quantiles_used:
loss += utils.tensorflow_quantile_loss(
a[Ellipsis, output_size * i:output_size * (i + 1)],
b[Ellipsis, output_size * i:output_size * (i + 1)], quantile)
return loss
