Update README.md

README.md

@@ -10,18 +10,27 @@ pipeline_tag: text-classification
 inital commit
 
 
+# Prediction
+
+The model outputs a Tensor of length 4.
+The table connects the position of the predicted probability to its dimension.
+
+| **Index** | **Dimension**            |
+|-----------|--------------------------|
+| 0         | Anti-Elitism             |
+| 1         | People-Centrism          |
+| 2         | Left-Wing Host-Ideology  |
+| 3         | Right-Wing Host-Ideology |
+
+
 # Performance
 
-This table presents the classification report for a 5-fold
-  cross-validation of our model. The hyperparameters are consistent
-  across all 5 runs. The final and published model was then trained on
-  all data with the same hyperparameters. It is evident that the model
-  performs, on average, best for anti-elitism but performs the worst for
-  the detection of right-wing host ideology. The relatively small
-  standard deviations suggest that the split into training and test data
-  has minimal impact on model performance. Therefore, it is expected
-  that the performance of the final model will be comparable to what is
-  depicted here.
+This table presents the classification report for a 5-fold cross-validation of our model.
+The hyperparameters are consistent across all 5 runs. The final and published model was then trained on all data with the same hyperparameters.
+It is evident that the model performs, on average, best for anti-elitism but performs the worst for detecting right-wing host ideology.
+The relatively small standard deviations suggest that the split into training and test data has minimal impact on model performance. 
+Therefore, it is expected that the performance of the final model will be comparable to what is
+depicted here.
 
 | Dimension           | Precision     | Recall        | F1            |
 |---------------------|---------------|---------------|---------------|
@@ -29,6 +38,6 @@ This table presents the classification report for a 5-fold
 | People-Centrism     | 0.670 (0.011) | 0.725 (0.040) | 0.696 (0.019) |
 | Left-Wing Ideology  | 0.664 (0.023) | 0.771 (0.024) | 0.713 (0.010) |
 | Right-Wing Ideology | 0.654 (0.029) | 0.698 (0.050) | 0.674 (0.031) |
-|                     |               |               |               |
+| ---                 | ---           | ---           | ---           |
 | micro avg           | 0.732 (0.009) | 0.805 (0.006) | 0.767 (0.007) |
 | macro avg           | 0.700 (0.011) | 0.770 (0.010) | 0.733 (0.010) |