sgdBERT (previously named "sdg-classification-bert"), is an NLP model for classifying text with respect to the United Nations sustainable development goals (SDG).
This text classification model was developed by fine-tuning the bert-base-uncased pre-trained model. The training data for this fine-tuned model was sourced from the publicly available OSDG Community Dataset (OSDG-CD) Version 2023.10 at
https://zenodo.org/records/8397907.
This model was made as part of academic research at Deakin University. The goal was to make a transformer-based SDG text classification model that anyone could use. Only the first 16 UN SDGs supported. The primary model details are highlighted below:
This is a fine-tuned model and therefore requires no further training.
Use the code below to get started with the model.
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2tokenizer = AutoTokenizer.from_pretrained("sadickam/sdg-classification-bert")
3model = AutoModelForSequenceClassification.from_pretrained("sadickam/sdg-classification-bert")
The training data includes text from a wide range of industries and academic research fields. Hence, this fine-tuned model is not for a specific industry.
If you use this model, please cite the journal article article below.