Update README.md

README.md

@@ -60,13 +60,13 @@ The classifier was trained on 500,000 pairs of code files and their scores from
 We added a classification head with a single regression output to StarEncoder and trained the model for 20 epochs with a learning rate of 3e-4. During training, the embedding and encoder layers were frozen to focus on the classification head. 
 
 It achieves the following results on the evaluation set:
-- Loss: 0.4719
-- Precision: 0.6074
-- Recall: 0.4618
-- F1 Macro: 0.4864
-- Accuracy: 0.5478
-- F1 Binary Minimum3: 0.8854
-- F1 Binary Minimum2: 0.9418
+- Loss: 0.3043
+- Precision: 0.6007
+- Recall: 0.5268
+- F1 Macro: 0.5255
+- Accuracy: 0.6694
+- F1 Binary Minimum3: 0.8552
+- F1 Binary Minimum2: 0.9316
 
 While the macro F1 scores across the 1-5 rating scale are relatively low due to the model's difficulty in distinguishing between higher-rated samples, the classifier performs well for our primary filtering task. When converting to binary classification, using a threshold of 2 achieves the F1 scores ranges between 0.8 and 0.9 for most Stack-Edu  classifiers, whereas a threshold of 3 yields F1 scores between 0.5 and 0.8. With the Highest being Python, SQL, C, Rust and the lowest being TypeScript and PHP.