Browse Source

Fix name of f1_score.

tags/v0.100.4-load-saved-model
Haiping Chen 2 years ago
parent
commit
9726fc6f81
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Keras/Metrics/IMetricsApi.cs
  2. +1
    -1
      src/TensorFlowNET.Keras/Metrics/MetricsApi.cs

+ 1
- 1
src/TensorFlowNET.Core/Keras/Metrics/IMetricsApi.cs View File

@@ -78,7 +78,7 @@ public interface IMetricsApi
IMetricFunc F1Score(int num_classes, IMetricFunc F1Score(int num_classes,
string? average = null, string? average = null,
float threshold = -1f, float threshold = -1f,
string name = "fbeta_score",
string name = "f1_score",
TF_DataType dtype = TF_DataType.TF_FLOAT); TF_DataType dtype = TF_DataType.TF_FLOAT);


/// <summary> /// <summary>


+ 1
- 1
src/TensorFlowNET.Keras/Metrics/MetricsApi.cs View File

@@ -86,7 +86,7 @@
public IMetricFunc CosineSimilarity(string name = "cosine_similarity", TF_DataType dtype = TF_DataType.TF_FLOAT, Axis? axis = null) public IMetricFunc CosineSimilarity(string name = "cosine_similarity", TF_DataType dtype = TF_DataType.TF_FLOAT, Axis? axis = null)
=> new CosineSimilarity(name: name, dtype: dtype, axis: axis ?? -1); => new CosineSimilarity(name: name, dtype: dtype, axis: axis ?? -1);


public IMetricFunc F1Score(int num_classes, string? average = null, float threshold = -1, string name = "fbeta_score", TF_DataType dtype = TF_DataType.TF_FLOAT)
public IMetricFunc F1Score(int num_classes, string? average = null, float threshold = -1, string name = "f1_score", TF_DataType dtype = TF_DataType.TF_FLOAT)
=> new F1Score(num_classes, average: average, threshold: threshold, name: name, dtype: dtype); => new F1Score(num_classes, average: average, threshold: threshold, name: name, dtype: dtype);


public IMetricFunc FBetaScore(int num_classes, string? average = null, float beta = 0.1F, float threshold = -1, string name = "fbeta_score", TF_DataType dtype = TF_DataType.TF_FLOAT) public IMetricFunc FBetaScore(int num_classes, string? average = null, float beta = 0.1F, float threshold = -1, string name = "fbeta_score", TF_DataType dtype = TF_DataType.TF_FLOAT)


Loading…
Cancel
Save