Hugging Face's logo

philipobiorah
/
bert-imdb-model

Text Classification
Transformers
PyTorch
Safetensors
English
bert
code
sentiment-analysis
imdb
nlp
Model card Files
xet
 Community
1
philipobiorah commited on
Commit
6cafa2b
·
verified ·
1 Parent(s): 25b777d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -30,7 +30,7 @@ This repository contains a fine-tuned BERT model for sentiment analysis on IMDb
30
  - **Labels**:
31
  - `0`: Negative
32
  - `1`: Positive
33
- <!-- ## Evaluation
34
  | **Model** | **SST-2 Accuracy** | **Yelp Accuracy** | **Amazon Accuracy** | **IMDB Accuracy** |
35
  |-----------------------------------|------------------|------------------|------------------|------------------|
36
  | **philipobiorah/bert-imdb-model** | **0.89** | **0.89** | **0.89** | **0.96** |
@@ -38,7 +38,7 @@ This repository contains a fine-tuned BERT model for sentiment analysis on IMDb
38
  | **RoBERTa-Sentiment** | **0.40** | **0.42** | **0.47** | **0.79** |
39
  | **Logistic Regression** | **0.83** | **0.91** | **0.86** | **0.85** |
40
  | **Naive Bayes** | **0.77** | **0.86** | **0.84** | **0.85** |
41
- -->
42
  ## Usage
43
  ### **Load the Model in Python**
44
  ```python
 
30
  - **Labels**:
31
  - `0`: Negative
32
  - `1`: Positive
33
+ ## Evaluation
34
  | **Model** | **SST-2 Accuracy** | **Yelp Accuracy** | **Amazon Accuracy** | **IMDB Accuracy** |
35
  |-----------------------------------|------------------|------------------|------------------|------------------|
36
  | **philipobiorah/bert-imdb-model** | **0.89** | **0.89** | **0.89** | **0.96** |
 
38
  | **RoBERTa-Sentiment** | **0.40** | **0.42** | **0.47** | **0.79** |
39
  | **Logistic Regression** | **0.83** | **0.91** | **0.86** | **0.85** |
40
  | **Naive Bayes** | **0.77** | **0.86** | **0.84** | **0.85** |
41
+
42
  ## Usage
43
  ### **Load the Model in Python**
44
  ```python