Upload folder using huggingface_hub

training_checkpoints/checkpoint-8359/config.json

@@ -0,0 +1,78 @@
+{
+  "_name_or_path": "xlnet/xlnet-base-cased",
+  "architectures": [
+    "XLNetForSequenceClassification"
+  ],
+  "attn_type": "bi",
+  "bi_data": false,
+  "bos_token_id": 1,
+  "clamp_len": -1,
+  "d_head": 64,
+  "d_inner": 3072,
+  "d_model": 768,
+  "dropout": 0.1,
+  "end_n_top": 5,
+  "eos_token_id": 2,
+  "ff_activation": "gelu",
+  "id2label": {
+    "0": "Issue",
+    "1": "Court Discourse",
+    "2": "Conclusion",
+    "3": "Precedent Analysis",
+    "4": "Section Analysis",
+    "5": "Argument by Petitioner",
+    "6": "Fact",
+    "7": "Argument by Respondent",
+    "8": "Ratio",
+    "9": "Appellant",
+    "10": "Respondent",
+    "11": "Argument by Appellant",
+    "12": "Petitioner",
+    "13": "Judge",
+    "14": "Argument by Defendant"
+  },
+  "initializer_range": 0.02,
+  "label2id": {
+    "Appellant": 9,
+    "Argument by Appellant": 11,
+    "Argument by Defendant": 14,
+    "Argument by Petitioner": 5,
+    "Argument by Respondent": 7,
+    "Conclusion": 2,
+    "Court Discourse": 1,
+    "Fact": 6,
+    "Issue": 0,
+    "Judge": 13,
+    "Petitioner": 12,
+    "Precedent Analysis": 3,
+    "Ratio": 8,
+    "Respondent": 10,
+    "Section Analysis": 4
+  },
+  "layer_norm_eps": 1e-12,
+  "mem_len": null,
+  "model_type": "xlnet",
+  "n_head": 12,
+  "n_layer": 12,
+  "pad_token_id": 5,
+  "problem_type": "single_label_classification",
+  "reuse_len": null,
+  "same_length": false,
+  "start_n_top": 5,
+  "summary_activation": "tanh",
+  "summary_last_dropout": 0.1,
+  "summary_type": "last",
+  "summary_use_proj": true,
+  "task_specific_params": {
+    "text-generation": {
+      "do_sample": true,
+      "max_length": 250
+    }
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.35.2",
+  "untie_r": true,
+  "use_mems_eval": true,
+  "use_mems_train": false,
+  "vocab_size": 32000
+}

training_checkpoints/checkpoint-8359/model.safetensors

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

training_checkpoints/checkpoint-8359/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62783473cd943cf804e6b750b2520198b1f766f4da1a6638deeb65bba340e593
+size 938733178

training_checkpoints/checkpoint-8359/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99dcb52d38813ccf92fd4505b4b60bb47f00977ee773683fdd34df3d42969236
+size 14244

training_checkpoints/checkpoint-8359/scheduler.pt

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

training_checkpoints/checkpoint-8359/special_tokens_map.json

@@ -0,0 +1,19 @@
+{
+  "additional_special_tokens": [
+    "<eop>",
+    "<eod>"
+  ],
+  "bos_token": "<s>",
+  "cls_token": "<cls>",
+  "eos_token": "</s>",
+  "mask_token": {
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<pad>",
+  "sep_token": "<sep>",
+  "unk_token": "<unk>"
+}

training_checkpoints/checkpoint-8359/spiece.model

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

training_checkpoints/checkpoint-8359/tokenizer_config.json

@@ -0,0 +1,94 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<cls>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "<sep>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "5": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "6": {
+      "content": "<mask>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "7": {
+      "content": "<eod>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "8": {
+      "content": "<eop>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<eop>",
+    "<eod>"
+  ],
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "<cls>",
+  "do_lower_case": false,
+  "eos_token": "</s>",
+  "keep_accents": false,
+  "mask_token": "<mask>",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "remove_space": true,
+  "sep_token": "<sep>",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "XLNetTokenizer",
+  "unk_token": "<unk>"
+}

training_checkpoints/checkpoint-8359/trainer_state.json

@@ -0,0 +1,401 @@
+{
+  "best_metric": 0.7706785605409691,
+  "best_model_checkpoint": "xlnet-base-cased/checkpoint-7716",
+  "epoch": 13.0,
+  "eval_steps": 500,
+  "global_step": 8359,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.78,
+      "learning_rate": 4.871695178849145e-05,
+      "loss": 1.2613,
+      "step": 500
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7676219984508134,
+      "eval_f1_macro": 0.4818756771742453,
+      "eval_f1_micro": 0.7676219984508135,
+      "eval_f1_weighted": 0.7524385212708887,
+      "eval_loss": 0.775787353515625,
+      "eval_macro_fpr": 0.022019799606126652,
+      "eval_macro_sensitivity": 0.5128622254580788,
+      "eval_macro_specificity": 0.9823717840367622,
+      "eval_precision": 0.7672637518290021,
+      "eval_precision_macro": 0.5269420024270713,
+      "eval_recall": 0.7676219984508134,
+      "eval_recall_macro": 0.5128622254580788,
+      "eval_runtime": 65.0712,
+      "eval_samples_per_second": 19.84,
+      "eval_steps_per_second": 2.49,
+      "eval_weighted_fpr": 0.021165514321998025,
+      "eval_weighted_sensitivity": 0.7676219984508134,
+      "eval_weighted_specificity": 0.9679547621006213,
+      "step": 643
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 4.7420943494038366e-05,
+      "loss": 0.7364,
+      "step": 1000
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.8071262587141751,
+      "eval_f1_macro": 0.701878478303997,
+      "eval_f1_micro": 0.8071262587141751,
+      "eval_f1_weighted": 0.8012788916547595,
+      "eval_loss": 0.6754755973815918,
+      "eval_macro_fpr": 0.01739858713969909,
+      "eval_macro_sensitivity": 0.6971761972643127,
+      "eval_macro_specificity": 0.9854830758512745,
+      "eval_precision": 0.8087721671110009,
+      "eval_precision_macro": 0.7425135911700289,
+      "eval_recall": 0.8071262587141751,
+      "eval_recall_macro": 0.6971761972643127,
+      "eval_runtime": 86.774,
+      "eval_samples_per_second": 14.878,
+      "eval_steps_per_second": 1.867,
+      "eval_weighted_fpr": 0.016782368403316036,
+      "eval_weighted_sensitivity": 0.8071262587141751,
+      "eval_weighted_specificity": 0.9751198790549438,
+      "step": 1286
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 4.612493519958528e-05,
+      "loss": 0.6021,
+      "step": 1500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.8063516653756778,
+      "eval_f1_macro": 0.722931923188587,
+      "eval_f1_micro": 0.8063516653756778,
+      "eval_f1_weighted": 0.8013697737227732,
+      "eval_loss": 0.8442708253860474,
+      "eval_macro_fpr": 0.01757802934678216,
+      "eval_macro_sensitivity": 0.7262014539622089,
+      "eval_macro_specificity": 0.9852110484486459,
+      "eval_precision": 0.8016312862866737,
+      "eval_precision_macro": 0.7269969408457856,
+      "eval_recall": 0.8063516653756778,
+      "eval_recall_macro": 0.7262014539622089,
+      "eval_runtime": 89.8454,
+      "eval_samples_per_second": 14.369,
+      "eval_steps_per_second": 1.803,
+      "eval_weighted_fpr": 0.01686454398273071,
+      "eval_weighted_sensitivity": 0.8063516653756778,
+      "eval_weighted_specificity": 0.9718140613540133,
+      "step": 1929
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 4.4828926905132196e-05,
+      "loss": 0.5435,
+      "step": 2000
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 4.353551062726802e-05,
+      "loss": 0.4361,
+      "step": 2500
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.8001549186676995,
+      "eval_f1_macro": 0.7051095861561788,
+      "eval_f1_micro": 0.8001549186676995,
+      "eval_f1_weighted": 0.7970808578587494,
+      "eval_loss": 0.8849796056747437,
+      "eval_macro_fpr": 0.017973543700983616,
+      "eval_macro_sensitivity": 0.704754303632947,
+      "eval_macro_specificity": 0.984883284344937,
+      "eval_precision": 0.8001040615771999,
+      "eval_precision_macro": 0.716655759638538,
+      "eval_recall": 0.8001549186676995,
+      "eval_recall_macro": 0.704754303632947,
+      "eval_runtime": 88.029,
+      "eval_samples_per_second": 14.666,
+      "eval_steps_per_second": 1.84,
+      "eval_weighted_fpr": 0.017527173913043478,
+      "eval_weighted_sensitivity": 0.8001549186676995,
+      "eval_weighted_specificity": 0.9730943465063553,
+      "step": 2572
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 4.223950233281493e-05,
+      "loss": 0.3359,
+      "step": 3000
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.801704105344694,
+      "eval_f1_macro": 0.6458763793901992,
+      "eval_f1_micro": 0.801704105344694,
+      "eval_f1_weighted": 0.7961730342415405,
+      "eval_loss": 1.126444935798645,
+      "eval_macro_fpr": 0.01805997413754306,
+      "eval_macro_sensitivity": 0.6681366041180199,
+      "eval_macro_specificity": 0.9849908327913178,
+      "eval_precision": 0.7981459541145973,
+      "eval_precision_macro": 0.6531140613829887,
+      "eval_recall": 0.801704105344694,
+      "eval_recall_macro": 0.6681366041180199,
+      "eval_runtime": 90.8127,
+      "eval_samples_per_second": 14.216,
+      "eval_steps_per_second": 1.784,
+      "eval_weighted_fpr": 0.0173606401736064,
+      "eval_weighted_sensitivity": 0.801704105344694,
+      "eval_weighted_specificity": 0.973158386525071,
+      "step": 3215
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 4.0943494038361847e-05,
+      "loss": 0.2827,
+      "step": 3500
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.7993803253292022,
+      "eval_f1_macro": 0.7042383323872946,
+      "eval_f1_micro": 0.7993803253292022,
+      "eval_f1_weighted": 0.7951558065494785,
+      "eval_loss": 1.1471006870269775,
+      "eval_macro_fpr": 0.01833120632003448,
+      "eval_macro_sensitivity": 0.692160115457962,
+      "eval_macro_specificity": 0.9845339432297603,
+      "eval_precision": 0.8091520232110664,
+      "eval_precision_macro": 0.7389074945114087,
+      "eval_recall": 0.7993803253292022,
+      "eval_recall_macro": 0.692160115457962,
+      "eval_runtime": 89.506,
+      "eval_samples_per_second": 14.424,
+      "eval_steps_per_second": 1.81,
+      "eval_weighted_fpr": 0.017610661589719183,
+      "eval_weighted_sensitivity": 0.7993803253292022,
+      "eval_weighted_specificity": 0.9686288231172028,
+      "step": 3858
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 3.964748574390876e-05,
+      "loss": 0.2626,
+      "step": 4000
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 3.835147744945568e-05,
+      "loss": 0.1945,
+      "step": 4500
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.814872192099148,
+      "eval_f1_macro": 0.7667275846579111,
+      "eval_f1_micro": 0.814872192099148,
+      "eval_f1_weighted": 0.812156980007075,
+      "eval_loss": 1.1841164827346802,
+      "eval_macro_fpr": 0.016569799909363286,
+      "eval_macro_sensitivity": 0.7598143558862378,
+      "eval_macro_specificity": 0.9859673331635007,
+      "eval_precision": 0.8128843369221012,
+      "eval_precision_macro": 0.7850186311075981,
+      "eval_recall": 0.814872192099148,
+      "eval_recall_macro": 0.7598143558862378,
+      "eval_runtime": 89.1729,
+      "eval_samples_per_second": 14.477,
+      "eval_steps_per_second": 1.817,
+      "eval_weighted_fpr": 0.015968463954032203,
+      "eval_weighted_sensitivity": 0.814872192099148,
+      "eval_weighted_specificity": 0.974637805353365,
+      "step": 4501
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 3.705546915500259e-05,
+      "loss": 0.1286,
+      "step": 5000
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.8079008520526724,
+      "eval_f1_macro": 0.728267610383121,
+      "eval_f1_micro": 0.8079008520526724,
+      "eval_f1_weighted": 0.8067061599230408,
+      "eval_loss": 1.3231384754180908,
+      "eval_macro_fpr": 0.01714064720569306,
+      "eval_macro_sensitivity": 0.7216321376051643,
+      "eval_macro_specificity": 0.9855732952526467,
+      "eval_precision": 0.8105008603379217,
+      "eval_precision_macro": 0.763038482726323,
+      "eval_recall": 0.8079008520526724,
+      "eval_recall_macro": 0.7216321376051643,
+      "eval_runtime": 93.4716,
+      "eval_samples_per_second": 13.812,
+      "eval_steps_per_second": 1.733,
+      "eval_weighted_fpr": 0.0167003367003367,
+      "eval_weighted_sensitivity": 0.8079008520526724,
+      "eval_weighted_specificity": 0.9756985767370285,
+      "step": 5144
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 3.576205287713841e-05,
+      "loss": 0.1304,
+      "step": 5500
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.8102246320681642,
+      "eval_f1_macro": 0.7569552546114842,
+      "eval_f1_micro": 0.8102246320681642,
+      "eval_f1_weighted": 0.808840893500485,
+      "eval_loss": 1.3869109153747559,
+      "eval_macro_fpr": 0.017055465242864547,
+      "eval_macro_sensitivity": 0.7602881393177737,
+      "eval_macro_specificity": 0.9856184320870693,
+      "eval_precision": 0.811846243162855,
+      "eval_precision_macro": 0.7704913288832418,
+      "eval_recall": 0.8102246320681642,
+      "eval_recall_macro": 0.7602881393177737,
+      "eval_runtime": 88.7368,
+      "eval_samples_per_second": 14.549,
+      "eval_steps_per_second": 1.826,
+      "eval_weighted_fpr": 0.016455101081335213,
+      "eval_weighted_sensitivity": 0.8102246320681642,
+      "eval_weighted_specificity": 0.9740518492378762,
+      "step": 5787
+    },
+    {
+      "epoch": 9.33,
+      "learning_rate": 3.4466044582685334e-05,
+      "loss": 0.0875,
+      "step": 6000
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.7823392718822618,
+      "eval_f1_macro": 0.7192346851166359,
+      "eval_f1_micro": 0.7823392718822618,
+      "eval_f1_weighted": 0.7816836829450283,
+      "eval_loss": 1.690091609954834,
+      "eval_macro_fpr": 0.01985842187936775,
+      "eval_macro_sensitivity": 0.7020361329217759,
+      "eval_macro_specificity": 0.9833585449617043,
+      "eval_precision": 0.7931868181724199,
+      "eval_precision_macro": 0.7600588253545632,
+      "eval_recall": 0.7823392718822618,
+      "eval_recall_macro": 0.7020361329217759,
+      "eval_runtime": 88.4203,
+      "eval_samples_per_second": 14.601,
+      "eval_steps_per_second": 1.832,
+      "eval_weighted_fpr": 0.019485472574717427,
+      "eval_weighted_sensitivity": 0.7823392718822618,
+      "eval_weighted_specificity": 0.9680389025433043,
+      "step": 6430
+    },
+    {
+      "epoch": 10.11,
+      "learning_rate": 3.317003628823225e-05,
+      "loss": 0.1088,
+      "step": 6500
+    },
+    {
+      "epoch": 10.89,
+      "learning_rate": 3.187402799377916e-05,
+      "loss": 0.1075,
+      "step": 7000
+    },
+    {
+      "epoch": 11.0,
+      "eval_accuracy": 0.7978311386522076,
+      "eval_f1_macro": 0.7470393699186935,
+      "eval_f1_micro": 0.7978311386522076,
+      "eval_f1_weighted": 0.793529238632398,
+      "eval_loss": 1.6517162322998047,
+      "eval_macro_fpr": 0.01832601410957563,
+      "eval_macro_sensitivity": 0.7566567183934103,
+      "eval_macro_specificity": 0.9849074579220182,
+      "eval_precision": 0.8021421826960871,
+      "eval_precision_macro": 0.7513026067404466,
+      "eval_recall": 0.7978311386522076,
+      "eval_recall_macro": 0.7566567183934103,
+      "eval_runtime": 91.1007,
+      "eval_samples_per_second": 14.171,
+      "eval_steps_per_second": 1.778,
+      "eval_weighted_fpr": 0.017778080512226686,
+      "eval_weighted_sensitivity": 0.7978311386522076,
+      "eval_weighted_specificity": 0.9757807301780667,
+      "step": 7073
+    },
+    {
+      "epoch": 11.66,
+      "learning_rate": 3.057801969932608e-05,
+      "loss": 0.0632,
+      "step": 7500
+    },
+    {
+      "epoch": 12.0,
+      "eval_accuracy": 0.814872192099148,
+      "eval_f1_macro": 0.7706785605409691,
+      "eval_f1_micro": 0.814872192099148,
+      "eval_f1_weighted": 0.814999517133868,
+      "eval_loss": 1.528985857963562,
+      "eval_macro_fpr": 0.016687805753048134,
+      "eval_macro_sensitivity": 0.7772039784617947,
+      "eval_macro_specificity": 0.9859096029272495,
+      "eval_precision": 0.8184480381747821,
+      "eval_precision_macro": 0.7745990994843823,
+      "eval_recall": 0.814872192099148,
+      "eval_recall_macro": 0.7772039784617947,
+      "eval_runtime": 89.9659,
+      "eval_samples_per_second": 14.35,
+      "eval_steps_per_second": 1.801,
+      "eval_weighted_fpr": 0.015968463954032203,
+      "eval_weighted_sensitivity": 0.814872192099148,
+      "eval_weighted_specificity": 0.9737718518095959,
+      "step": 7716
+    },
+    {
+      "epoch": 12.44,
+      "learning_rate": 2.92846034214619e-05,
+      "loss": 0.0565,
+      "step": 8000
+    },
+    {
+      "epoch": 13.0,
+      "eval_accuracy": 0.8063516653756778,
+      "eval_f1_macro": 0.7536745391279389,
+      "eval_f1_micro": 0.8063516653756778,
+      "eval_f1_weighted": 0.806105583642158,
+      "eval_loss": 1.5766452550888062,
+      "eval_macro_fpr": 0.017353934865295607,
+      "eval_macro_sensitivity": 0.7627918284863161,
+      "eval_macro_specificity": 0.9855508104644062,
+      "eval_precision": 0.8106650569621797,
+      "eval_precision_macro": 0.7527919670677403,
+      "eval_recall": 0.8063516653756778,
+      "eval_recall_macro": 0.7627918284863161,
+      "eval_runtime": 87.0044,
+      "eval_samples_per_second": 14.838,
+      "eval_steps_per_second": 1.862,
+      "eval_weighted_fpr": 0.01686454398273071,
+      "eval_weighted_sensitivity": 0.8063516653756778,
+      "eval_weighted_specificity": 0.9769104915904122,
+      "step": 8359
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 19290,
+  "num_train_epochs": 30,
+  "save_steps": 500,
+  "total_flos": 1.9030344688502784e+16,
+  "trial_name": null,
+  "trial_params": null
+}

training_checkpoints/checkpoint-8359/training_args.bin

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