Downloads · 30 days
5
16% of all-time downloads
dveytia/distilbert_ORO_Branch
distilbert_ORO_Branch is a text classification model from dveytia. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
This model is a fine-tuned version of distilbert-base-uncased to classify an article's text (title + abstract + keywords). The intention is that this model will be used AFTER establishing an article's relevance to an…
Downloads · 30 days
5
16% of all-time downloads
All-time downloads
31
Public
Repo size
268 MB
Likes
0
Public
Click a slice to open those files.
.h5268 MB · 100%
From the Hugging Face model README
This model is a fine-tuned version of distilbert-base-uncased to classify an article's text (title + abstract + keywords). The intention is that this model will be used AFTER establishing an article's relevance to an Ocean-related option (ORO) (see the screening model card on huggingface). This model will then classify a relevant article futher into the type of ORO: Mitigation, Natural resilence or Societal adaptation.
It achieves the following results on the evaluation set:
This model predicts for relevance to three labels specifying the type of ocean related option as a value between 0 and 1. Therefore a number > 0.5 indicates it is more likely to be relevant that that type of ORO.
This model is intended to be applied to article text (title + abstract + keywords) retrieved from citation indexed databases such as Web of Science or Scopus using a search query. This can be used to autonomously classify relevant articles from a large volume of literature and can be used in analyses that provide a granular map of the distribution of relevant studies.
For a description of the dataset, see the paper (in prep.)
The following hyperparameters were used during training:
| Train Loss | Train Binary Accuracy | Validation Loss | Validation Binary Accuracy | Epoch |
|---|---|---|---|---|
| 0.4863 | 0.7650 | 0.4002 | 0.8368 | 0 |
| 0.2445 | 0.9238 | 0.2649 | 0.8993 | 1 |
| 0.1604 | 0.9509 | 0.2817 | 0.8946 | 2 |