Hugging Face's logo

nasa-impact
/
tdamm-classification

Text Classification
Transformers
Safetensors
English
bert
Text Classification
TDAMM
Multi-label Classification
NASA
Astrophysics
Model card Files Community
bishwaspraveen commited on
Commit
b60e4c7
·
verified ·
1 Parent(s): 1f9a904

Updated label to id mapping in config

Browse files
Files changed (1) hide show
  1. config.json +138 -91
config.json CHANGED
@@ -1,103 +1,150 @@
1
  {
2
- "_name_or_path": "adsabs/astroBERT",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "classifier_dropout": null,
8
- "gradient_checkpointing": false,
9
- "hidden_act": "gelu",
10
- "hidden_dropout_prob": 0.1,
11
- "hidden_size": 768,
12
  "id2label": {
13
- "0": "LABEL_0",
14
- "1": "LABEL_1",
15
- "2": "LABEL_2",
16
- "3": "LABEL_3",
17
- "4": "LABEL_4",
18
- "5": "LABEL_5",
19
- "6": "LABEL_6",
20
- "7": "LABEL_7",
21
- "8": "LABEL_8",
22
- "9": "LABEL_9",
23
- "10": "LABEL_10",
24
- "11": "LABEL_11",
25
- "12": "LABEL_12",
26
- "13": "LABEL_13",
27
- "14": "LABEL_14",
28
- "15": "LABEL_15",
29
- "16": "LABEL_16",
30
- "17": "LABEL_17",
31
- "18": "LABEL_18",
32
- "19": "LABEL_19",
33
- "20": "LABEL_20",
34
- "21": "LABEL_21",
35
- "22": "LABEL_22",
36
- "23": "LABEL_23",
37
- "24": "LABEL_24",
38
- "25": "LABEL_25",
39
- "26": "LABEL_26",
40
- "27": "LABEL_27",
41
- "28": "LABEL_28",
42
- "29": "LABEL_29",
43
- "30": "LABEL_30",
44
- "31": "LABEL_31",
45
- "32": "LABEL_32",
46
- "33": "LABEL_33",
47
- "34": "LABEL_34",
48
- "35": "LABEL_35"
49
  },
50
- "initializer_range": 0.02,
51
- "intermediate_size": 3072,
52
  "label2id": {
53
- "LABEL_0": 0,
54
- "LABEL_1": 1,
55
- "LABEL_10": 10,
56
- "LABEL_11": 11,
57
- "LABEL_12": 12,
58
- "LABEL_13": 13,
59
- "LABEL_14": 14,
60
- "LABEL_15": 15,
61
- "LABEL_16": 16,
62
- "LABEL_17": 17,
63
- "LABEL_18": 18,
64
- "LABEL_19": 19,
65
- "LABEL_2": 2,
66
- "LABEL_20": 20,
67
- "LABEL_21": 21,
68
- "LABEL_22": 22,
69
- "LABEL_23": 23,
70
- "LABEL_24": 24,
71
- "LABEL_25": 25,
72
- "LABEL_26": 26,
73
- "LABEL_27": 27,
74
- "LABEL_28": 28,
75
- "LABEL_29": 29,
76
- "LABEL_3": 3,
77
- "LABEL_30": 30,
78
- "LABEL_31": 31,
79
- "LABEL_32": 32,
80
- "LABEL_33": 33,
81
- "LABEL_34": 34,
82
- "LABEL_35": 35,
83
- "LABEL_4": 4,
84
- "LABEL_5": 5,
85
- "LABEL_6": 6,
86
- "LABEL_7": 7,
87
- "LABEL_8": 8,
88
- "LABEL_9": 9
89
  },
90
- "layer_norm_eps": 1e-12,
91
- "max_position_embeddings": 512,
92
- "model_type": "bert",
93
- "num_attention_heads": 12,
94
- "num_hidden_layers": 12,
95
  "pad_token_id": 0,
96
- "position_embedding_type": "absolute",
 
 
 
97
  "problem_type": "multi_label_classification",
98
- "torch_dtype": "float32",
99
- "transformers_version": "4.48.1",
 
 
 
 
 
 
 
 
 
 
 
100
  "type_vocab_size": 2,
 
 
 
101
  "use_cache": true,
102
- "vocab_size": 30000
103
- }
 
1
  {
2
+ "return_dict": true,
3
+ "output_hidden_states": false,
4
+ "output_attentions": false,
5
+ "torchscript": false,
6
+ "torch_dtype": "float32",
7
+ "use_bfloat16": false,
8
+ "tf_legacy_loss": false,
9
+ "pruned_heads": {},
10
+ "tie_word_embeddings": true,
11
+ "chunk_size_feed_forward": 0,
12
+ "is_encoder_decoder": false,
13
+ "is_decoder": false,
14
+ "cross_attention_hidden_size": null,
15
+ "add_cross_attention": false,
16
+ "tie_encoder_decoder": false,
17
+ "max_length": 20,
18
+ "min_length": 0,
19
+ "do_sample": false,
20
+ "early_stopping": false,
21
+ "num_beams": 1,
22
+ "num_beam_groups": 1,
23
+ "diversity_penalty": 0.0,
24
+ "temperature": 1.0,
25
+ "top_k": 50,
26
+ "top_p": 1.0,
27
+ "typical_p": 1.0,
28
+ "repetition_penalty": 1.0,
29
+ "length_penalty": 1.0,
30
+ "no_repeat_ngram_size": 0,
31
+ "encoder_no_repeat_ngram_size": 0,
32
+ "bad_words_ids": null,
33
+ "num_return_sequences": 1,
34
+ "output_scores": false,
35
+ "return_dict_in_generate": false,
36
+ "forced_bos_token_id": null,
37
+ "forced_eos_token_id": null,
38
+ "remove_invalid_values": false,
39
+ "exponential_decay_length_penalty": null,
40
+ "suppress_tokens": null,
41
+ "begin_suppress_tokens": null,
42
  "architectures": [
43
  "BertForSequenceClassification"
44
  ],
45
+ "finetuning_task": null,
 
 
 
 
 
46
  "id2label": {
47
+ "0": "Active Galactic Nuclei",
48
+ "1": "Binary Black Holes",
49
+ "2": "Binary Neutron Stars",
50
+ "3": "Binary Pulsars",
51
+ "4": "Burst",
52
+ "5": "Cataclysmic Variables",
53
+ "6": "Compact Binary Inspiral",
54
+ "7": "Continuous",
55
+ "8": "Cosmic Rays",
56
+ "9": "Exoplanets",
57
+ "10": "Fast Blue Optical Transients",
58
+ "11": "Fast Radio Bursts",
59
+ "12": "Gamma rays",
60
+ "13": "Gamma-ray Bursts",
61
+ "14": "Infrared",
62
+ "15": "Intermediate Mass",
63
+ "16": "Kilonovae",
64
+ "17": "Magnetars",
65
+ "18": "Neutrinos",
66
+ "19": "Neutron Star-Black Hole",
67
+ "20": "Novae",
68
+ "21": "Optical",
69
+ "22": "Pevatrons",
70
+ "23": "Pulsar Wind Nebulae",
71
+ "24": "Pulsars",
72
+ "25": "Radio",
73
+ "26": "Stellar Mass",
74
+ "27": "Stellar flares",
75
+ "28": "Stochastic",
76
+ "29": "SuperNovae",
77
+ "30": "Supermassive",
78
+ "31": "Supernova Remnants",
79
+ "32": "Ultraviolet",
80
+ "33": "White Dwarf Binaries",
81
+ "34": "X-rays",
82
+ "35": "non-TDAMM"
83
  },
 
 
84
  "label2id": {
85
+ "Active Galactic Nuclei": 0,
86
+ "Binary Black Holes": 1,
87
+ "Binary Neutron Stars": 2,
88
+ "Binary Pulsars": 3,
89
+ "Burst": 4,
90
+ "Cataclysmic Variables": 5,
91
+ "Compact Binary Inspiral": 6,
92
+ "Continuous": 7,
93
+ "Cosmic Rays": 8,
94
+ "Exoplanets": 9,
95
+ "Fast Blue Optical Transients": 10,
96
+ "Fast Radio Bursts": 11,
97
+ "Gamma rays": 12,
98
+ "Gamma-ray Bursts": 13,
99
+ "Infrared": 14,
100
+ "Intermediate Mass": 15,
101
+ "Kilonovae": 16,
102
+ "Magnetars": 17,
103
+ "Neutrinos": 18,
104
+ "Neutron Star-Black Hole": 19,
105
+ "Novae": 20,
106
+ "Optical": 21,
107
+ "Pevatrons": 22,
108
+ "Pulsar Wind Nebulae": 23,
109
+ "Pulsars": 24,
110
+ "Radio": 25,
111
+ "Stellar Mass": 26,
112
+ "Stellar flares": 27,
113
+ "Stochastic": 28,
114
+ "SuperNovae": 29,
115
+ "Supermassive": 30,
116
+ "Supernova Remnants": 31,
117
+ "Ultraviolet": 32,
118
+ "White Dwarf Binaries": 33,
119
+ "X-rays": 34,
120
+ "non-TDAMM": 35
121
  },
122
+ "tokenizer_class": null,
123
+ "prefix": null,
124
+ "bos_token_id": null,
 
 
125
  "pad_token_id": 0,
126
+ "eos_token_id": null,
127
+ "sep_token_id": null,
128
+ "decoder_start_token_id": null,
129
+ "task_specific_params": null,
130
  "problem_type": "multi_label_classification",
131
+ "_attn_implementation_autoset": true,
132
+ "transformers_version": "4.48.0",
133
+ "gradient_checkpointing": false,
134
+ "model_type": "bert",
135
+ "vocab_size": 30000,
136
+ "hidden_size": 768,
137
+ "num_hidden_layers": 12,
138
+ "num_attention_heads": 12,
139
+ "hidden_act": "gelu",
140
+ "intermediate_size": 3072,
141
+ "hidden_dropout_prob": 0.1,
142
+ "attention_probs_dropout_prob": 0.1,
143
+ "max_position_embeddings": 512,
144
  "type_vocab_size": 2,
145
+ "initializer_range": 0.02,
146
+ "layer_norm_eps": 1e-12,
147
+ "position_embedding_type": "absolute",
148
  "use_cache": true,
149
+ "classifier_dropout": null
150
+ }