Tobias commited on
Commit
1a0fad4
1 Parent(s): 67fbb5c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -16
README.md CHANGED
@@ -15,19 +15,5 @@ widget:
15
  example_title: "Example 4"
16
  ---
17
 
18
- # German Hotel Review Sentiment Classification
19
- A model trained on English Hotel Reviews from Switzerland. The base model is the [bert-base-uncased](https://huggingface.co/bert-base-uncased). The last hidden layer of the base model was extracted and a classification layer was added. The entire model was then trained for 5 epochs on our dataset.
20
-
21
- # Model Performance
22
-
23
- | Classes | Precision | Recall | F1 Score |
24
- | :---: | :---: | :---: |:---: |
25
- | Positive | 86.36% | 95.00% | 90.48% |
26
- | Negative | 90.91% | 76.92% | 83.33% |
27
- | Neutral | 50.00% | 42.86% | 46.15% |
28
- | Accuracy | | | 83.00% |
29
- | Macro Average | 75.76% | 71.59% | 73.32% |
30
- | Weighted Average | 82.45% | 83.00% | 82.41% |
31
-
32
- ## Confusion Matrix
33
- ![Confusion Matrix](bert-base-uncased_English_sentiment.jpg)
15
  example_title: "Example 4"
16
  ---
17
 
18
+ # English Hotel Review Sentiment Classification
19
+ A model trained on English Hotel Reviews from Switzerland. The base model is the [bert-base-uncased](https://huggingface.co/bert-base-uncased). The last hidden layer of the base model was extracted and a classification layer was added. The entire model was then trained for 5 epochs on our dataset.