Upload 9 files

MultiModel_Epochs10_bespoke_results.json

@@ -0,0 +1,64 @@
+{
+    "epoch": 10.0,
+    "eval_accuracy": 0.8538622129436325,
+    "eval_auc_pe_Present": 0.9684379023558478,
+    "eval_auc_rv_dil_Present": 0.9715198237885462,
+    "eval_auc_Normal_or_No_Label": 0.988722656692735,
+    "eval_auc_tricuspid_valve_native_regurgitation_Present": 0.9994070480425847,
+    "eval_auc_wma_Present": 0.9927310231023101,
+    "eval_auc_lv_dil_Present": 0.9841185342915446,
+    "eval_auc_aortic_valve_native_stenosis_Present": 0.9984241962129877,
+    "eval_auc_mitral_valve_native_regurgitation_Present": 0.9949279328589673,
+    "eval_auc_lv_sys_func_Present": 0.9936147229849783,
+    "eval_auc_rv_sys_func_Present": 0.9858074338197734,
+    "eval_auc_aortic_valve_native_regurgitation_Present": 0.99593653453422,
+    "eval_auc_lv_dias_func_Present": 0.9922497900830007,
+    "eval_f1_pe_Present": 0.8571428571428571,
+    "eval_f1_rv_dil_Present": 0.8041237113402062,
+    "eval_f1_Normal_or_No_Label": 0.9624277456647399,
+    "eval_f1_tricuspid_valve_native_regurgitation_Present": 0.9642857142857143,
+    "eval_f1_wma_Present": 0.9166666666666666,
+    "eval_f1_lv_dil_Present": 0.9171270718232044,
+    "eval_f1_aortic_valve_native_stenosis_Present": 0.9156626506024096,
+    "eval_f1_mitral_valve_native_regurgitation_Present": 0.9505494505494505,
+    "eval_f1_lv_sys_func_Present": 0.9750415973377704,
+    "eval_f1_rv_sys_func_Present": 0.8974358974358975,
+    "eval_f1_aortic_valve_native_regurgitation_Present": 0.9688888888888889,
+    "eval_f1_lv_dias_func_Present": 0.9384164222873901,
+    "eval_f1_macro": 0.9223140561687662,
+    "eval_f1_weighted": 0.9425007148312412,
+    "eval_loss": 0.05956393480300903,
+    "eval_prec_pe_Present": 0.84375,
+    "eval_prec_rv_dil_Present": 0.8297872340425532,
+    "eval_prec_Normal_or_No_Label": 0.9624277456647399,
+    "eval_prec_tricuspid_valve_native_regurgitation_Present": 0.9759036144578314,
+    "eval_prec_wma_Present": 0.9565217391304348,
+    "eval_prec_lv_dil_Present": 0.9222222222222223,
+    "eval_prec_aortic_valve_native_stenosis_Present": 0.95,
+    "eval_prec_mitral_valve_native_regurgitation_Present": 0.9153439153439153,
+    "eval_prec_lv_sys_func_Present": 0.979933110367893,
+    "eval_prec_rv_sys_func_Present": 0.9130434782608695,
+    "eval_prec_aortic_valve_native_regurgitation_Present": 0.9732142857142857,
+    "eval_prec_lv_dias_func_Present": 0.9411764705882353,
+    "eval_prec_macro": 0.9302769846494151,
+    "eval_prec_weighted": 0.9472192429644785,
+    "eval_recall_pe_Present": 0.8709677419354839,
+    "eval_recall_rv_dil_Present": 0.78,
+    "eval_recall_Normal_or_No_Label": 0.9624277456647399,
+    "eval_recall_tricuspid_valve_native_regurgitation_Present": 0.9529411764705882,
+    "eval_recall_wma_Present": 0.88,
+    "eval_recall_lv_dil_Present": 0.9120879120879121,
+    "eval_recall_aortic_valve_native_stenosis_Present": 0.8837209302325582,
+    "eval_recall_mitral_valve_native_regurgitation_Present": 0.9885714285714285,
+    "eval_recall_lv_sys_func_Present": 0.9701986754966887,
+    "eval_recall_rv_sys_func_Present": 0.8823529411764706,
+    "eval_recall_aortic_valve_native_regurgitation_Present": 0.9646017699115044,
+    "eval_recall_lv_dias_func_Present": 0.935672514619883,
+    "eval_recall_macro": 0.9152952363472715,
+    "eval_recall_weighted": 0.9385441527446301,
+    "eval_roc_auc_macro": 0.9888247998972913,
+    "eval_roc_auc_weighted": 0.9909024741692697,
+    "eval_runtime": 5.19,
+    "eval_samples_per_second": 184.586,
+    "eval_steps_per_second": 18.497
+}

config.json

@@ -0,0 +1,57 @@
+{
+  "_name_or_path": "CLTL/MedRoBERTa.nl",
+  "architectures": [
+    "RobertaForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "classifier_dropout": null,
+  "eos_token_id": 2,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "pe_Present",
+    "1": "rv_dil_Present",
+    "2": "wma_Present",
+    "3": "lv_dil_Present",
+    "4": "aortic_valve_native_stenosis_Present",
+    "5": "mitral_valve_native_regurgitation_Present",
+    "6": "lv_sys_func_Present",
+    "7": "rv_sys_func_Present",
+    "8": "aortic_valve_native_regurgitation_Present",
+    "9": "lv_dias_func_Present",
+    "10": "Normal_or_No_Label",
+    "11": "tricuspid_valve_native_regurgitation_Present"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "Normal_or_No_Label": 10,
+    "aortic_valve_native_regurgitation_Present": 8,
+    "aortic_valve_native_stenosis_Present": 4,
+    "lv_dias_func_Present": 9,
+    "lv_dil_Present": 3,
+    "lv_sys_func_Present": 6,
+    "mitral_valve_native_regurgitation_Present": 5,
+    "pe_Present": 0,
+    "rv_dil_Present": 1,
+    "rv_sys_func_Present": 7,
+    "tricuspid_valve_native_regurgitation_Present": 11,
+    "wma_Present": 2
+  },
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 514,
+  "model_type": "roberta",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 1,
+  "position_embedding_type": "absolute",
+  "problem_type": "multi_label_classification",
+  "torch_dtype": "float32",
+  "transformers_version": "4.40.0",
+  "type_vocab_size": 1,
+  "use_cache": true,
+  "vocab_size": 52000
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83d66c798aec0e26ff3505763e71701add3887f5c993d925f44fee8f46c4d30c
+size 503973504

special_tokens_map.json

@@ -0,0 +1,51 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "cls_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,57 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "<mask>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "errors": "replace",
+  "mask_token": "<mask>",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "sep_token": "</s>",
+  "tokenizer_class": "RobertaTokenizer",
+  "trim_offsets": true,
+  "unk_token": "<unk>"
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4518296a63557578b0be8c5bc99f04c8c5ad54b4f97af4501094f52ca6af1325
+size 5048