Terry Zhang
commited on
Commit
·
108eb1c
1
Parent(s):
7688055
add climate bert
Browse files- tasks/text.py +2 -1
tasks/text.py
CHANGED
|
@@ -23,6 +23,7 @@ models_descriptions = {
|
|
| 23 |
"baseline": "random baseline",
|
| 24 |
"tfidf_xgb": "TF-IDF vectorizer and XGBoost classifier",
|
| 25 |
"bert_base_pruned": "Pruned BERT base model",
|
|
|
|
| 26 |
}
|
| 27 |
|
| 28 |
|
|
@@ -108,7 +109,7 @@ def bert_classifier(test_dataset: dict, model: str):
|
|
| 108 |
|
| 109 |
@router.post(ROUTE, tags=["Text Task"])
|
| 110 |
async def evaluate_text(request: TextEvaluationRequest,
|
| 111 |
-
model: str = "
|
| 112 |
"""
|
| 113 |
Evaluate text classification for climate disinformation detection.
|
| 114 |
|
|
|
|
| 23 |
"baseline": "random baseline",
|
| 24 |
"tfidf_xgb": "TF-IDF vectorizer and XGBoost classifier",
|
| 25 |
"bert_base_pruned": "Pruned BERT base model",
|
| 26 |
+
'climate_bert_pruned': "Fine-tuned and pruned DistilRoBERTa pre-trained on climate texts"
|
| 27 |
}
|
| 28 |
|
| 29 |
|
|
|
|
| 109 |
|
| 110 |
@router.post(ROUTE, tags=["Text Task"])
|
| 111 |
async def evaluate_text(request: TextEvaluationRequest,
|
| 112 |
+
model: str = "climate_bert_pruned"):
|
| 113 |
"""
|
| 114 |
Evaluate text classification for climate disinformation detection.
|
| 115 |
|