[skip actions] [ml] 2024-07-23T12:00:46+03:00

Samsung · Jul 23, 2024 · 69f8c33 · 69f8c33
1 parent 6ff7ae7
commit 69f8c33
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/credsweeper/ml_model/ml_model.onnx b/credsweeper/ml_model/ml_model.onnx
diff --git a/experiment/main.py b/experiment/main.py
@@ -126,7 +126,7 @@ def main(cred_data_location: str, jobs: int) -> str:
     y_test = get_y_labels(df_test)
     print(f"Class-1 prop on test: {np.mean(y_test):.4f}")
 
-    max_epochs = 100
+    max_epochs = 1
     # ^^^ the line is patched in GitHub action to speed-up test train
     batch_size = 2048
     early_stopping = EarlyStopping(monitor="val_loss", patience=7, mode="min", restore_best_weights=True, verbose=1)