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 |
|