Terry Zhang commited on
Commit
108eb1c
·
1 Parent(s): 7688055

add climate bert

Browse files
Files changed (1) hide show
  1. 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 = "bert_base_pruned"):
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