Browse Source

Update ILossFunc.cs

pull/681/head
dataangel GitHub 5 years ago
parent
commit
b73a6fea32
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Keras/Losses/ILossFunc.cs

+ 2
- 2
src/TensorFlowNET.Keras/Losses/ILossFunc.cs View File

@@ -1,8 +1,8 @@
namespace Tensorflow.Keras.Losses
namespace Tensorflow.Keras.Losses
{ {
public interface ILossFunc public interface ILossFunc
{ {
string Reduction { get; } string Reduction { get; }
Tensor Call(Tensor y_true, Tensor y_pred);
Tensor Call(Tensor y_true, Tensor y_pred, Tensor sample_weight = null);
} }
} }

Loading…
Cancel
Save