loubnabnl HF Staff commited on
Commit
992a82a
·
verified ·
1 Parent(s): 105a89d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -70,7 +70,14 @@ It achieves the following results on the evaluation set:
70
  - Accuracy: 0.5471
71
  - F1 Binary Minimum3: 0.7028
72
 
73
- 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.55 and 0.70. With the Highest being C and Python and the lowest being JavaScript and PhP.
 
 
 
 
 
 
 
74
  ### Training hyperparameters
75
 
76
  The following hyperparameters were used during training:
 
70
  - Accuracy: 0.5471
71
  - F1 Binary Minimum3: 0.7028
72
 
73
+ 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.55 and 0.70. With the Highest being Python, C, Rust and the lowest being TypeScript and PHP.
74
+
75
+ <div style="display: flex; justify-content: center; gap: 20px;">
76
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/61c141342aac764ce1654e43/z0lCnH2nqYMu09JsxPvFb.png" width="600">
77
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/61c141342aac764ce1654e43/W6BQO6Rkafpj-96fWdCOv.png" width="600">
78
+ </div>
79
+ We validated these classifiers by filtering Stack v2 data and testing on an intermediate SmolLM2 checkpoint. Filtering with a threshold of 3 improved performance across most languages while maintaining adequate data volume, though Java showed better results with a threshold of 2.
80
+
81
  ### Training hyperparameters
82
 
83
  The following hyperparameters were used during training: