Update BertClassification.cs

Beacontownfc · Oceania2018 · commit 12ef6288378f · 2023-08-27T19:51:10.000-05:00
diff --git a/src/TensorFlowNET.Examples/NaturalLanguageProcessing/BertClassification.cs b/src/TensorFlowNET.Examples/NaturalLanguageProcessing/BertClassification.cs
@@ -15,10 +15,10 @@ namespace TensorFlowNET.Examples
 {
     class BertClassification : SciSharpExample, IExample
     {
-        int max_seq_len = 180;
-        int batch_size = 4;
+        int max_seq_len = 512;
+        int batch_size = 32;
         int num_classes = 2;
-        int epoch = 3;
+        int epoch = 10;
         float learning_rate = (float)2e-5;
         string pretrained_weight_path = "./tf_model.h5";
         BertConfig config = new BertConfig();

Original file line number	Diff line number	Diff line change
`@@ -15,10 +15,10 @@ namespace TensorFlowNET.Examples`
`15`	`15`	`{`
`16`	`16`	`class BertClassification : SciSharpExample, IExample`
`17`	`17`	`{`
`18`		`- int max_seq_len = 180;`
`19`		`- int batch_size = 4;`
	`18`	`+ int max_seq_len = 512;`
	`19`	`+ int batch_size = 32;`
`20`	`20`	`int num_classes = 2;`
`21`		`- int epoch = 3;`
	`21`	`+ int epoch = 10;`
`22`	`22`	`float learning_rate = (float)2e-5;`
`23`	`23`	`string pretrained_weight_path = "./tf_model.h5";`
`24`	`24`	`BertConfig config = new BertConfig();`