elam2909 commited on
Commit
ad22bbf
·
verified ·
1 Parent(s): 572246f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ This model is intended for:
29
  - **Base model:** `bert-base-uncased`
30
  - **Fine-tuned on:** Disaster Tweet Classification Dataset (binary classification: disaster vs. not disaster)
31
  - **Framework:** Hugging Face Transformers + PyTorch
32
- - **Classifier Head:** A linear layer on top of BERT for classification
33
 
34
 
35
  ## Training Dataset
 
29
  - **Base model:** `bert-base-uncased`
30
  - **Fine-tuned on:** Disaster Tweet Classification Dataset (binary classification: disaster vs. not disaster)
31
  - **Framework:** Hugging Face Transformers + PyTorch
32
+ - **Post-BERT Classifier:** Random Forest trained on BERT [CLS] embeddings
33
 
34
 
35
  ## Training Dataset