Text Classification
Transformers
Safetensors
English
roberta
Text Classification
TDAMM
Multi-label Classification
NASA
Astrophysics
Science Document Entity
text-embeddings-inference
Instructions to use nasa-impact/tdamm-classification-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use nasa-impact/tdamm-classification-v2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="nasa-impact/tdamm-classification-v2")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("nasa-impact/tdamm-classification-v2") model = AutoModelForSequenceClassification.from_pretrained("nasa-impact/tdamm-classification-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "architectures": [ | |
| "RobertaForSequenceClassification" | |
| ], | |
| "attention_probs_dropout_prob": 0.1, | |
| "bos_token_id": 0, | |
| "classifier_dropout": null, | |
| "eos_token_id": 2, | |
| "hidden_act": "gelu", | |
| "hidden_dropout_prob": 0.1, | |
| "hidden_size": 768, | |
| "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" | |
| }, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 3072, | |
| "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 | |
| }, | |
| "layer_norm_eps": 1e-05, | |
| "max_position_embeddings": 1026, | |
| "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.50.0.dev0", | |
| "type_vocab_size": 1, | |
| "use_cache": true, | |
| "vocab_size": 50265 | |
| } | |