NikkeS commited on
Commit
b5705c3
·
verified ·
1 Parent(s): ec69f53

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -7
README.md CHANGED
@@ -85,8 +85,9 @@ print(predict_sentiment("The acting was terrible, and the story made no sense.")
85
  ## Training Details
86
 
87
  ### Training Data
88
- - The model was fine-tuned on the **IMDB dataset** (50,000 labeled movie reviews).
89
- - The dataset is **balanced** (25,000 positive and 25,000 negative reviews).
 
90
 
91
  ### Training Procedure
92
  #### Preprocessing
@@ -108,14 +109,14 @@ print(predict_sentiment("The acting was terrible, and the story made no sense.")
108
 
109
  ### Testing Data, Factors & Metrics
110
  #### Testing Data
111
- - The model was evaluated on a **10,000-sample test set** from the IMDB dataset.
112
 
113
  #### Metrics
114
- - **Accuracy:** 92,4%
115
  - **Precision, Recall, F1-score:**
116
- - **Precision:** 92,4%
117
- - **Recall:** 92.3%
118
- - **F1-score:** 92.3%
119
 
120
  ## Model Examination
121
  - The model performs well on **general sentiment classification** but may struggle with **sarcasm, irony, or very short reviews**.
 
85
  ## Training Details
86
 
87
  ### Training Data
88
+ - The model was fine-tuned on the IMDB dataset (50,000 labeled movie reviews).
89
+ - The dataset is balanced (25,000 positive and 25,000 negative reviews).
90
+ - The training split consisted of 40,000 samples, while 5,000 samples were used for validation.
91
 
92
  ### Training Procedure
93
  #### Preprocessing
 
109
 
110
  ### Testing Data, Factors & Metrics
111
  #### Testing Data
112
+ - The model was evaluated on a 5,000-sample test set from the IMDB dataset.
113
 
114
  #### Metrics
115
+ - **Accuracy:** 90,4%
116
  - **Precision, Recall, F1-score:**
117
+ - **Precision:** 92,1%
118
+ - **Recall:** 88.2%
119
+ - **F1-score:** 90.0%
120
 
121
  ## Model Examination
122
  - The model performs well on **general sentiment classification** but may struggle with **sarcasm, irony, or very short reviews**.