Updates to DS

Update DS with addtl. HIS samples

BatchSize_32_Metrics.txt

@@ -1,7 +1,7 @@
-Date:  11-06, 19:24
+Date:  11-07, 15:18
 Batch Size: 32
 Eval Strategy: no
 Early Stopping: steps
 Early Stopping Patience: 5
 Best Model Metric: None
-{'eval_loss': 0.2943166196346283, 'eval_accuracy': 0.8874773139745916, 'eval_f1': 0.8851851851851852, 'eval_runtime': 9.486, 'eval_samples_per_second': 58.086, 'eval_steps_per_second': 1.898, 'epoch': 12.76}
+{'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

@@ -8,7 +8,15 @@
   "dim": 768,
   "dropout": 0.1,
   "hidden_dim": 3072,
+  "id2label": {
+    "0": "Low Risk",
+    "1": "High Risk"
+  },
   "initializer_range": 0.02,
+  "label2id": {
+    "High Risk": 1,
+    "Low Risk": 0
+  },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:38550c3f11db0d2fe10e422448355db727c781f9ad434d1bf0ff9684a727f91c
+oid sha256:204c66b0bd4c4bb15ad7b66de2520b0440b56f000d70ade541cb9ddf4e3906c4
 size 267854125