Updated weights
Browse files- pytorch_model.bin +1 -1
- training_config.json +4 -4
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 954851171
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a5b872ccafa0a00340fe5f02f64cdf7684427269ae6645a688313492499b1d9e
|
| 3 |
size 954851171
|
training_config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"model_name": "michiyasunaga/BioLinkBERT-base",
|
| 3 |
-
"hub_model_id": "chrisvoncsefalvay/
|
| 4 |
"mimicause_repo": "chrisvoncsefalvay/mimicause-private",
|
| 5 |
"batch_size": 16,
|
| 6 |
"gradient_accumulation_steps": 4,
|
|
@@ -28,7 +28,7 @@
|
|
| 28 |
"Other"
|
| 29 |
],
|
| 30 |
"seed": 42,
|
| 31 |
-
"best_f1": 0.
|
| 32 |
-
"best_auc": 0.
|
| 33 |
-
"optimal_threshold": 0.
|
| 34 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"model_name": "michiyasunaga/BioLinkBERT-base",
|
| 3 |
+
"hub_model_id": "chrisvoncsefalvay/CRAG-dual-encoder-mimicause",
|
| 4 |
"mimicause_repo": "chrisvoncsefalvay/mimicause-private",
|
| 5 |
"batch_size": 16,
|
| 6 |
"gradient_accumulation_steps": 4,
|
|
|
|
| 28 |
"Other"
|
| 29 |
],
|
| 30 |
"seed": 42,
|
| 31 |
+
"best_f1": 0.988861985472155,
|
| 32 |
+
"best_auc": 0.998112678527832,
|
| 33 |
+
"optimal_threshold": 0.702573835849762
|
| 34 |
}
|