digitalhealth-healthyliving
/

MediFlow

Text Classification

Inference Endpoints

Model card Files Files and versions Community

eperezs commited on Jul 15

Commit

c6c3c5c

•

1 Parent(s): 409e8a2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ print(result)
 - **learning_rate**: 2e-5
 - **batch_size**: 4
-- **num_train_epochs**: 2
 - **weight_decay**: 0.015
 - **optimizer**: AdamW
 - **test_size**: 0.2

 - **learning_rate**: 2e-5
 - **batch_size**: 4
+- **num_train_epochs**: 3
 - **weight_decay**: 0.015
 - **optimizer**: AdamW
 - **test_size**: 0.2