Text Classification
Transformers
PyTorch
English
roberta
NASA
GES-DISC
Earth Sciences
text-embeddings-inference
Instructions to use arminmehrabian/nasa-impact-nasa-smd-ibm-st-v2-classification-finetuned with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use arminmehrabian/nasa-impact-nasa-smd-ibm-st-v2-classification-finetuned with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="arminmehrabian/nasa-impact-nasa-smd-ibm-st-v2-classification-finetuned")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("arminmehrabian/nasa-impact-nasa-smd-ibm-st-v2-classification-finetuned") model = AutoModelForSequenceClassification.from_pretrained("arminmehrabian/nasa-impact-nasa-smd-ibm-st-v2-classification-finetuned", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Model Information
This model is based on the nasa-impact/nasa-smd-ibm-st-v2 model. It has been fine-tuned for a classification task using approximately 1400 publication abstracts in earth sciences. The model was validated on 1000 unseen abstracts. The labelling of the publications was conducted by the NASA GES-DISC science team. The training and validation dataset, along with performance scores, will be published here soon.
Label Mapping
The model classifies inputs into the following categories, each represented by a unique label ID:
| Category | ID |
|---|---|
| Agriculture | 0 |
| Air Quality | 1 |
| Atmospheric/Ocean Indicators | 2 |
| Cryospheric Indicators | 3 |
| Droughts | 4 |
| Earthquakes | 5 |
| Ecosystems | 6 |
| Energy Production/Use | 7 |
| Environmental Impacts | 8 |
| Floods | 9 |
| Greenhouse Gases | 10 |
| Habitat Conversion/Fragmentation | 11 |
| Heat | 12 |
| Land Surface/Agriculture Indicators | 13 |
| Public Health | 14 |
| Severe Storms | 15 |
| Sun-Earth Interactions | 16 |
| Validation | 17 |
| Volcanic Eruptions | 18 |
| Water Quality | 19 |
| Wildfires | 20 |
- Downloads last month
- 9