From 5b4d497910ee5b638052537c68cb0400fc56dc32 Mon Sep 17 00:00:00 2001 From: dataangel Date: Mon, 14 Dec 2020 20:37:56 +0800 Subject: [PATCH] Update LossesApi.cs --- src/TensorFlowNET.Keras/Losses/LossesApi.cs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Keras/Losses/LossesApi.cs b/src/TensorFlowNET.Keras/Losses/LossesApi.cs index 3e66b395..2078c8ce 100644 --- a/src/TensorFlowNET.Keras/Losses/LossesApi.cs +++ b/src/TensorFlowNET.Keras/Losses/LossesApi.cs @@ -1,4 +1,4 @@ -namespace Tensorflow.Keras.Losses +namespace Tensorflow.Keras.Losses { public class LossesApi { @@ -7,5 +7,8 @@ public ILossFunc CategoricalCrossentropy(bool from_logits = false) => new CategoricalCrossentropy(from_logits: from_logits); + + public ILossFunc MeanSquaredError(string reduction = null) + => new MeanSquaredError(reduction: reduction); } }