Browse Source

fix: revise earlystopping callback's min_delta parameter

tags/v0.110.4-Transformer-Model
Wanglongzhi2001 2 years ago
parent
commit
fff5029b02
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Keras/Callbacks/Earlystopping.cs

+ 2
- 2
src/TensorFlowNET.Keras/Callbacks/Earlystopping.cs View File

@@ -11,7 +11,7 @@ namespace Tensorflow.Keras.Callbacks;
public class EarlyStopping: ICallback public class EarlyStopping: ICallback
{ {
int _paitence; int _paitence;
int _min_delta;
float _min_delta;
int _verbose; int _verbose;
int _stopped_epoch; int _stopped_epoch;
int _wait; int _wait;
@@ -26,7 +26,7 @@ public class EarlyStopping: ICallback
CallbackParams _parameters; CallbackParams _parameters;
public Dictionary<string, List<float>>? history { get; set; } public Dictionary<string, List<float>>? history { get; set; }
// user need to pass a CallbackParams to EarlyStopping, CallbackParams at least need the model // user need to pass a CallbackParams to EarlyStopping, CallbackParams at least need the model
public EarlyStopping(CallbackParams parameters,string monitor = "val_loss", int min_delta = 0, int patience = 0,
public EarlyStopping(CallbackParams parameters,string monitor = "val_loss", float min_delta = 0f, int patience = 0,
int verbose = 1, string mode = "auto", float baseline = 0f, bool restore_best_weights = false, int verbose = 1, string mode = "auto", float baseline = 0f, bool restore_best_weights = false,
int start_from_epoch = 0) int start_from_epoch = 0)
{ {


Loading…
Cancel
Save