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); } }