End of training

README.md

@@ -1,10 +1,66 @@
----
-datasets:
-- arxiv-community/arxiv_dataset
-metrics:
-- accuracy
-base_model:
-- microsoft/deberta-v3-base
----
-# ArXiv paper category classifier
-DistilBERT model fine-tuned on the ArXiv dataset to predict the category of a given paper based on its title and abstract
+---
+library_name: transformers
+license: mit
+base_model: microsoft/deberta-v3-base
+tags:
+- generated_from_trainer
+metrics:
+- accuracy
+model-index:
+- name: bert-paper-classifier-arxiv
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# bert-paper-classifier-arxiv
+
+This model is a fine-tuned version of [microsoft/deberta-v3-base](https://huggingface.co/microsoft/deberta-v3-base) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 3.7652
+- Accuracy: 0.31
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-05
+- train_batch_size: 64
+- eval_batch_size: 8
+- seed: 42
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 128
+- optimizer: Use OptimizerNames.ADAFACTOR and the args are:
+No additional optimizer arguments
+- lr_scheduler_type: cosine
+- num_epochs: 2
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| No log        | 1.0   | 8    | 4.0299          | 0.31     |
+| No log        | 1.8   | 14   | 3.7652          | 0.31     |
+
+
+### Framework versions
+
+- Transformers 4.48.3
+- Pytorch 2.5.1+cu124
+- Datasets 3.5.0
+- Tokenizers 0.21.0

config.json

@@ -0,0 +1,292 @@
+{
+  "_name_or_path": "microsoft/deberta-v3-base",
+  "architectures": [
+    "DebertaV2ForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "adap-org",
+    "1": "astro-ph",
+    "2": "astro-ph.CO",
+    "3": "astro-ph.EP",
+    "4": "astro-ph.GA",
+    "5": "astro-ph.IM",
+    "6": "astro-ph.SR",
+    "7": "cmp-lg",
+    "8": "cond-mat",
+    "9": "cond-mat.dis-nn",
+    "10": "cond-mat.mtrl-sci",
+    "11": "cond-mat.other",
+    "12": "cond-mat.soft",
+    "13": "cond-mat.stat-mech",
+    "14": "cond-mat.supr-con",
+    "15": "cs.AI",
+    "16": "cs.AR",
+    "17": "cs.CC",
+    "18": "cs.CE",
+    "19": "cs.CG",
+    "20": "cs.CL",
+    "21": "cs.CR",
+    "22": "cs.CV",
+    "23": "cs.CY",
+    "24": "cs.DB",
+    "25": "cs.DC",
+    "26": "cs.DL",
+    "27": "cs.DM",
+    "28": "cs.DS",
+    "29": "cs.ET",
+    "30": "cs.FL",
+    "31": "cs.GL",
+    "32": "cs.GR",
+    "33": "cs.GT",
+    "34": "cs.HC",
+    "35": "cs.IR",
+    "36": "cs.IT",
+    "37": "cs.LG",
+    "38": "cs.LO",
+    "39": "cs.MA",
+    "40": "cs.MM",
+    "41": "cs.MS",
+    "42": "cs.NA",
+    "43": "cs.NE",
+    "44": "cs.NI",
+    "45": "cs.OH",
+    "46": "cs.OS",
+    "47": "cs.PF",
+    "48": "cs.PL",
+    "49": "cs.RO",
+    "50": "cs.SC",
+    "51": "cs.SD",
+    "52": "cs.SE",
+    "53": "cs.SI",
+    "54": "cs.SY",
+    "55": "econ.EM",
+    "56": "eess.AS",
+    "57": "eess.IV",
+    "58": "eess.SP",
+    "59": "gr-qc",
+    "60": "hep-ex",
+    "61": "hep-lat",
+    "62": "hep-ph",
+    "63": "hep-th",
+    "64": "math.AG",
+    "65": "math.AP",
+    "66": "math.AT",
+    "67": "math.CA",
+    "68": "math.CO",
+    "69": "math.CT",
+    "70": "math.DG",
+    "71": "math.DS",
+    "72": "math.FA",
+    "73": "math.GM",
+    "74": "math.GN",
+    "75": "math.GR",
+    "76": "math.GT",
+    "77": "math.HO",
+    "78": "math.LO",
+    "79": "math.MG",
+    "80": "math.NA",
+    "81": "math.NT",
+    "82": "math.OC",
+    "83": "math.PR",
+    "84": "math.RA",
+    "85": "math.RT",
+    "86": "math.ST",
+    "87": "nlin.AO",
+    "88": "nlin.CD",
+    "89": "nlin.CG",
+    "90": "nlin.PS",
+    "91": "nucl-th",
+    "92": "physics.ao-ph",
+    "93": "physics.bio-ph",
+    "94": "physics.chem-ph",
+    "95": "physics.class-ph",
+    "96": "physics.comp-ph",
+    "97": "physics.data-an",
+    "98": "physics.gen-ph",
+    "99": "physics.geo-ph",
+    "100": "physics.hist-ph",
+    "101": "physics.ins-det",
+    "102": "physics.med-ph",
+    "103": "physics.optics",
+    "104": "physics.soc-ph",
+    "105": "q-bio.BM",
+    "106": "q-bio.CB",
+    "107": "q-bio.GN",
+    "108": "q-bio.MN",
+    "109": "q-bio.NC",
+    "110": "q-bio.PE",
+    "111": "q-bio.QM",
+    "112": "q-bio.TO",
+    "113": "q-fin.CP",
+    "114": "q-fin.EC",
+    "115": "q-fin.GN",
+    "116": "q-fin.PM",
+    "117": "q-fin.RM",
+    "118": "q-fin.ST",
+    "119": "q-fin.TR",
+    "120": "quant-ph",
+    "121": "stat.AP",
+    "122": "stat.CO",
+    "123": "stat.ME",
+    "124": "stat.ML",
+    "125": "stat.OT"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "adap-org": 0,
+    "astro-ph": 1,
+    "astro-ph.CO": 2,
+    "astro-ph.EP": 3,
+    "astro-ph.GA": 4,
+    "astro-ph.IM": 5,
+    "astro-ph.SR": 6,
+    "cmp-lg": 7,
+    "cond-mat": 8,
+    "cond-mat.dis-nn": 9,
+    "cond-mat.mtrl-sci": 10,
+    "cond-mat.other": 11,
+    "cond-mat.soft": 12,
+    "cond-mat.stat-mech": 13,
+    "cond-mat.supr-con": 14,
+    "cs.AI": 15,
+    "cs.AR": 16,
+    "cs.CC": 17,
+    "cs.CE": 18,
+    "cs.CG": 19,
+    "cs.CL": 20,
+    "cs.CR": 21,
+    "cs.CV": 22,
+    "cs.CY": 23,
+    "cs.DB": 24,
+    "cs.DC": 25,
+    "cs.DL": 26,
+    "cs.DM": 27,
+    "cs.DS": 28,
+    "cs.ET": 29,
+    "cs.FL": 30,
+    "cs.GL": 31,
+    "cs.GR": 32,
+    "cs.GT": 33,
+    "cs.HC": 34,
+    "cs.IR": 35,
+    "cs.IT": 36,
+    "cs.LG": 37,
+    "cs.LO": 38,
+    "cs.MA": 39,
+    "cs.MM": 40,
+    "cs.MS": 41,
+    "cs.NA": 42,
+    "cs.NE": 43,
+    "cs.NI": 44,
+    "cs.OH": 45,
+    "cs.OS": 46,
+    "cs.PF": 47,
+    "cs.PL": 48,
+    "cs.RO": 49,
+    "cs.SC": 50,
+    "cs.SD": 51,
+    "cs.SE": 52,
+    "cs.SI": 53,
+    "cs.SY": 54,
+    "econ.EM": 55,
+    "eess.AS": 56,
+    "eess.IV": 57,
+    "eess.SP": 58,
+    "gr-qc": 59,
+    "hep-ex": 60,
+    "hep-lat": 61,
+    "hep-ph": 62,
+    "hep-th": 63,
+    "math.AG": 64,
+    "math.AP": 65,
+    "math.AT": 66,
+    "math.CA": 67,
+    "math.CO": 68,
+    "math.CT": 69,
+    "math.DG": 70,
+    "math.DS": 71,
+    "math.FA": 72,
+    "math.GM": 73,
+    "math.GN": 74,
+    "math.GR": 75,
+    "math.GT": 76,
+    "math.HO": 77,
+    "math.LO": 78,
+    "math.MG": 79,
+    "math.NA": 80,
+    "math.NT": 81,
+    "math.OC": 82,
+    "math.PR": 83,
+    "math.RA": 84,
+    "math.RT": 85,
+    "math.ST": 86,
+    "nlin.AO": 87,
+    "nlin.CD": 88,
+    "nlin.CG": 89,
+    "nlin.PS": 90,
+    "nucl-th": 91,
+    "physics.ao-ph": 92,
+    "physics.bio-ph": 93,
+    "physics.chem-ph": 94,
+    "physics.class-ph": 95,
+    "physics.comp-ph": 96,
+    "physics.data-an": 97,
+    "physics.gen-ph": 98,
+    "physics.geo-ph": 99,
+    "physics.hist-ph": 100,
+    "physics.ins-det": 101,
+    "physics.med-ph": 102,
+    "physics.optics": 103,
+    "physics.soc-ph": 104,
+    "q-bio.BM": 105,
+    "q-bio.CB": 106,
+    "q-bio.GN": 107,
+    "q-bio.MN": 108,
+    "q-bio.NC": 109,
+    "q-bio.PE": 110,
+    "q-bio.QM": 111,
+    "q-bio.TO": 112,
+    "q-fin.CP": 113,
+    "q-fin.EC": 114,
+    "q-fin.GN": 115,
+    "q-fin.PM": 116,
+    "q-fin.RM": 117,
+    "q-fin.ST": 118,
+    "q-fin.TR": 119,
+    "quant-ph": 120,
+    "stat.AP": 121,
+    "stat.CO": 122,
+    "stat.ME": 123,
+    "stat.ML": 124,
+    "stat.OT": 125
+  },
+  "layer_norm_eps": 1e-07,
+  "legacy": true,
+  "max_position_embeddings": 512,
+  "max_relative_positions": -1,
+  "model_type": "deberta-v2",
+  "norm_rel_ebd": "layer_norm",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "pooler_dropout": 0,
+  "pooler_hidden_act": "gelu",
+  "pooler_hidden_size": 768,
+  "pos_att_type": [
+    "p2c",
+    "c2p"
+  ],
+  "position_biased_input": false,
+  "position_buckets": 256,
+  "relative_attention": true,
+  "share_att_key": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.48.3",
+  "type_vocab_size": 0,
+  "vocab_size": 128100
+}

model.safetensors

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

spm.model

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