supaernova.steps.nflow.tf.loss
[docs]
module
supaernova.steps.nflow.tf.loss
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 | import os
from typing import TYPE_CHECKING
os.environ["TF_USE_LEGACY_KERAS"] = "1"
os.environ["KERAS_BACKEND"] = "tensorflow"
os.environ["TF_DETERMINISTIC_OPS"] = "1"
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
import tensorflow as tf
if TYPE_CHECKING:
from .tf import TFNFlowModel
def NegLogLikelihood(y_true, y_pred, *, model: "TFNFlowModel"):
return -y_pred
|