config.json

{
  "return_dict": true,
  "output_hidden_states": false,
  "output_attentions": false,
  "torchscript": false,
  "torch_dtype": "float32",
  "use_bfloat16": false,
  "tf_legacy_loss": false,
  "pruned_heads": {},
  "tie_word_embeddings": true,
  "chunk_size_feed_forward": 0,
  "is_encoder_decoder": false,
  "is_decoder": false,
  "cross_attention_hidden_size": null,
  "add_cross_attention": false,
  "tie_encoder_decoder": false,
  "max_length": 20,
  "min_length": 0,
  "do_sample": false,
  "early_stopping": false,
  "num_beams": 1,
  "num_beam_groups": 1,
  "diversity_penalty": 0.0,
  "temperature": 1.0,
  "top_k": 50,
  "top_p": 1.0,
  "typical_p": 1.0,
  "repetition_penalty": 1.0,
  "length_penalty": 1.0,
  "no_repeat_ngram_size": 0,
  "encoder_no_repeat_ngram_size": 0,
  "bad_words_ids": null,
  "num_return_sequences": 1,
  "output_scores": false,
  "return_dict_in_generate": false,
  "forced_bos_token_id": null,
  "forced_eos_token_id": null,
  "remove_invalid_values": false,
  "exponential_decay_length_penalty": null,
  "suppress_tokens": null,
  "begin_suppress_tokens": null,
  "architectures": [
    "BertForSequenceClassification"
  ],
  "finetuning_task": null,
  "id2label": {
    "0": "Active Galactic Nuclei",
    "1": "Binary Black Holes",
    "2": "Binary Neutron Stars",
    "3": "Binary Pulsars",
    "4": "Burst",
    "5": "Cataclysmic Variables",
    "6": "Compact Binary Inspiral",
    "7": "Continuous",
    "8": "Cosmic Rays",
    "9": "Exoplanets",
    "10": "Fast Blue Optical Transients",
    "11": "Fast Radio Bursts",
    "12": "Gamma rays",
    "13": "Gamma-ray Bursts",
    "14": "Infrared",
    "15": "Intermediate Mass",
    "16": "Kilonovae",
    "17": "Magnetars",
    "18": "Neutrinos",
    "19": "Neutron Star-Black Hole",
    "20": "Novae",
    "21": "Optical",
    "22": "Pevatrons",
    "23": "Pulsar Wind Nebulae",
    "24": "Pulsars",
    "25": "Radio",
    "26": "Stellar Mass",
    "27": "Stellar flares",
    "28": "Stochastic",
    "29": "SuperNovae",
    "30": "Supermassive",
    "31": "Supernova Remnants",
    "32": "Ultraviolet",
    "33": "White Dwarf Binaries",
    "34": "X-rays",
    "35": "non-TDAMM"
  },
  "label2id": {
    "Active Galactic Nuclei": 0,
    "Binary Black Holes": 1,
    "Binary Neutron Stars": 2,
    "Binary Pulsars": 3,
    "Burst": 4,
    "Cataclysmic Variables": 5,
    "Compact Binary Inspiral": 6,
    "Continuous": 7,
    "Cosmic Rays": 8,
    "Exoplanets": 9,
    "Fast Blue Optical Transients": 10,
    "Fast Radio Bursts": 11,
    "Gamma rays": 12,
    "Gamma-ray Bursts": 13,
    "Infrared": 14,
    "Intermediate Mass": 15,
    "Kilonovae": 16,
    "Magnetars": 17,
    "Neutrinos": 18,
    "Neutron Star-Black Hole": 19,
    "Novae": 20,
    "Optical": 21,
    "Pevatrons": 22,
    "Pulsar Wind Nebulae": 23,
    "Pulsars": 24,
    "Radio": 25,
    "Stellar Mass": 26,
    "Stellar flares": 27,
    "Stochastic": 28,
    "SuperNovae": 29,
    "Supermassive": 30,
    "Supernova Remnants": 31,
    "Ultraviolet": 32,
    "White Dwarf Binaries": 33,
    "X-rays": 34,
    "non-TDAMM": 35
  },
  "tokenizer_class": null,
  "prefix": null,
  "bos_token_id": null,
  "pad_token_id": 0,
  "eos_token_id": null,
  "sep_token_id": null,
  "decoder_start_token_id": null,
  "task_specific_params": null,
  "problem_type": "multi_label_classification",
  "_attn_implementation_autoset": true,
  "transformers_version": "4.48.0",
  "gradient_checkpointing": false,
  "model_type": "bert",
  "vocab_size": 30000,
  "hidden_size": 768,
  "num_hidden_layers": 12,
  "num_attention_heads": 12,
  "hidden_act": "gelu",
  "intermediate_size": 3072,
  "hidden_dropout_prob": 0.1,
  "attention_probs_dropout_prob": 0.1,
  "max_position_embeddings": 512,
  "type_vocab_size": 2,
  "initializer_range": 0.02,
  "layer_norm_eps": 1e-12,
  "position_embedding_type": "absolute",
  "use_cache": true,
  "classifier_dropout": null
}