Python tensorflow_quantile_loss Exemples

Exemple #1

Fichier : tft_model.py Projet : JeonJunHyuk/time-series-forecast-using-Transformer

                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

Exemple #2

        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