Updates to DS
Browse filesUpdate DS with addtl. HIS samples
- BatchSize_32_Metrics.txt +2 -2
- config.json +8 -0
- pytorch_model.bin +1 -1
- training_args.bin +0 -0
BatchSize_32_Metrics.txt
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
-
Date: 11-
|
2 |
Batch Size: 32
|
3 |
Eval Strategy: no
|
4 |
Early Stopping: steps
|
5 |
Early Stopping Patience: 5
|
6 |
Best Model Metric: None
|
7 |
-
{'eval_loss': 0.
|
|
|
1 |
+
Date: 11-07, 15:18
|
2 |
Batch Size: 32
|
3 |
Eval Strategy: no
|
4 |
Early Stopping: steps
|
5 |
Early Stopping Patience: 5
|
6 |
Best Model Metric: None
|
7 |
+
{'eval_loss': 0.29037001729011536, 'eval_accuracy': 0.8738898756660746, 'eval_f1': 0.8672897196261682, 'eval_runtime': 9.6195, 'eval_samples_per_second': 58.527, 'eval_steps_per_second': 1.871, 'epoch': 14.5}
|
config.json
CHANGED
@@ -8,7 +8,15 @@
|
|
8 |
"dim": 768,
|
9 |
"dropout": 0.1,
|
10 |
"hidden_dim": 3072,
|
|
|
|
|
|
|
|
|
11 |
"initializer_range": 0.02,
|
|
|
|
|
|
|
|
|
12 |
"max_position_embeddings": 512,
|
13 |
"model_type": "distilbert",
|
14 |
"n_heads": 12,
|
|
|
8 |
"dim": 768,
|
9 |
"dropout": 0.1,
|
10 |
"hidden_dim": 3072,
|
11 |
+
"id2label": {
|
12 |
+
"0": "Low Risk",
|
13 |
+
"1": "High Risk"
|
14 |
+
},
|
15 |
"initializer_range": 0.02,
|
16 |
+
"label2id": {
|
17 |
+
"High Risk": 1,
|
18 |
+
"Low Risk": 0
|
19 |
+
},
|
20 |
"max_position_embeddings": 512,
|
21 |
"model_type": "distilbert",
|
22 |
"n_heads": 12,
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 267854125
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:204c66b0bd4c4bb15ad7b66de2520b0440b56f000d70ade541cb9ddf4e3906c4
|
3 |
size 267854125
|
training_args.bin
CHANGED
Binary files a/training_args.bin and b/training_args.bin differ
|
|