dharshinig012 commited on
Commit
f301661
·
verified ·
1 Parent(s): a9d16bb

Upload train_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_config.json +17 -0
train_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "microsoft/graphcodebert-base",
3
+ "max_len": 512,
4
+ "batch_size": 16,
5
+ "accum_steps": 2,
6
+ "lr": 2e-05,
7
+ "seed": 42,
8
+ "epochs_ran": 10,
9
+ "best_val_f1": 0.9902200488997555,
10
+ "test_metrics": {
11
+ "accuracy": 0.9927095990279465,
12
+ "precision": 0.9901477832512315,
13
+ "recall": 0.995049504950495,
14
+ "f1": 0.9925925925925926,
15
+ "roc_auc": 0.999288144804934
16
+ }
17
+ }