Commit ·
a3d83fd
1
Parent(s): bb5629d
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,100 @@
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
-
# Aurora SDG
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
- nl
|
| 6 |
+
- de
|
| 7 |
+
- fr
|
| 8 |
+
- it
|
| 9 |
+
- is
|
| 10 |
+
- cs
|
| 11 |
+
- da
|
| 12 |
+
- es
|
| 13 |
+
- ca
|
| 14 |
+
metrics:
|
| 15 |
+
- accuracy
|
| 16 |
+
- matthews_correlation
|
| 17 |
+
pipeline_tag: text-classification
|
| 18 |
---
|
| 19 |
+
# Aurora SDG Multi-Label Multi-Class Model
|
| 20 |
+
|
| 21 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
| 22 |
+
This model is able to classify texts related to United Nations sustainable development goals (SDG) in multiple languages.
|
| 23 |
+
|
| 24 |
+

|
| 25 |
+
Source: https://sdgs.un.org/goals
|
| 26 |
+
|
| 27 |
+
## Model Details
|
| 28 |
+
|
| 29 |
+
### Model Description
|
| 30 |
+
|
| 31 |
+
<!-- Provide a longer summary of what this model is. -->
|
| 32 |
+
|
| 33 |
+
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) at https://zenodo.org/record/5550238#.ZBulfcJByF4.
|
| 34 |
+
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:
|
| 35 |
+
|
| 36 |
+
- **Model type:** Text classification
|
| 37 |
+
- **Language(s) (NLP):** English, Dutch, German, Icelandic, French, Czeck, Italian, Danisch, Spanish, Catalan
|
| 38 |
+
- **License:** cc-by-4.0
|
| 39 |
+
- **Finetuned from model [optional]:** bert-base-multilingual-uncased
|
| 40 |
+
|
| 41 |
+
### Model Sources
|
| 42 |
+
<!-- Provide the basic links for the model. -->
|
| 43 |
+
- **Repository:** option 1: https://huggingface.co/MauriceV2021/AuroraSDGsModel ; option 2 https://doi.org/10.5281/zenodo.7304546
|
| 44 |
+
- **Demo [optional]:** option 1: ; option 2: https://aurora-universities.eu/sdg-research/classify/
|
| 45 |
+
|
| 46 |
+
|
| 47 |
+
### Direct Use
|
| 48 |
+
|
| 49 |
+
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
| 50 |
+
|
| 51 |
+
This is a fine-tuned model and therefore requires no further training.
|
| 52 |
+
|
| 53 |
+
|
| 54 |
+
## How to Get Started with the Model
|
| 55 |
+
|
| 56 |
+
Use the code here to get started with the model: https://github.com/Aurora-Network-Global/sdgs_many_berts
|
| 57 |
+
|
| 58 |
+
|
| 59 |
+
## Training Data
|
| 60 |
+
|
| 61 |
+
<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
|
| 62 |
+
The training data includes text from 1.4 titles and abstracts of academic research papers, labeled with SDG Goals and Targets, according to an initial validated query.
|
| 63 |
+
|
| 64 |
+
See training data here: https://doi.org/10.5281/zenodo.5205672
|
| 65 |
+
|
| 66 |
+
### Evaluation of the Training data
|
| 67 |
+
|
| 68 |
+
- Avg_precision = 0.70
|
| 69 |
+
- Avg_recall = 0.15
|
| 70 |
+
|
| 71 |
+
Data evaluated by 244 domain expert senior researchers.
|
| 72 |
+
|
| 73 |
+
See evaluation report on the training data here: https://doi.org/10.5281/zenodo.4917107
|
| 74 |
+
|
| 75 |
+
|
| 76 |
+
## Training Hyperparameters
|
| 77 |
+
|
| 78 |
+
<!--
|
| 79 |
+
- Num_epoch = 3
|
| 80 |
+
- Learning rate = 5e-5
|
| 81 |
+
- Batch size = 16
|
| 82 |
+
-->
|
| 83 |
+
|
| 84 |
+
## Evaluation
|
| 85 |
+
|
| 86 |
+
#### Metrics
|
| 87 |
+
|
| 88 |
+
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
| 89 |
+
- Accuracy = 0.9
|
| 90 |
+
- Matthews correlation = 0.89
|
| 91 |
+
|
| 92 |
+
See evaluation report on the model here: https://doi.org/10.5281/zenodo.5603019
|
| 93 |
+
|
| 94 |
+
## Citation
|
| 95 |
+
Sadick, A.M. (2023). SDG classification with BERT. https://huggingface.co/sadickam/sdg-classification-bert
|
| 96 |
+
|
| 97 |
+
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
| 98 |
+
|
| 99 |
+
|
| 100 |
+
<!--## Model Card Contact -->
|