From b73a6fea32d02b7c53cfc5330bafa30cb2b7eaaa Mon Sep 17 00:00:00 2001 From: dataangel Date: Mon, 14 Dec 2020 21:59:58 +0800 Subject: [PATCH] Update ILossFunc.cs --- src/TensorFlowNET.Keras/Losses/ILossFunc.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/TensorFlowNET.Keras/Losses/ILossFunc.cs b/src/TensorFlowNET.Keras/Losses/ILossFunc.cs index 45c39dd2..4795461d 100644 --- a/src/TensorFlowNET.Keras/Losses/ILossFunc.cs +++ b/src/TensorFlowNET.Keras/Losses/ILossFunc.cs @@ -1,8 +1,8 @@ -namespace Tensorflow.Keras.Losses +namespace Tensorflow.Keras.Losses { public interface ILossFunc { string Reduction { get; } - Tensor Call(Tensor y_true, Tensor y_pred); + Tensor Call(Tensor y_true, Tensor y_pred, Tensor sample_weight = null); } }