Trained model with classification head weights

README.md

@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [allenai/scibert_scivocab_uncased](https://huggingface.co/allenai/scibert_scivocab_uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2972
-- Accuracy: 0.8919
+- Loss: 0.2965
+- Accuracy: 0.8886
 
 ## Model description
 
@@ -49,16 +49,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | Accuracy |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 1.3931        | 1.0   | 2124  | 0.6939          | 0.8322   |
-| 0.9757        | 2.0   | 4248  | 0.4680          | 0.8614   |
-| 0.8006        | 3.0   | 6372  | 0.4260          | 0.8680   |
-| 0.712         | 4.0   | 8496  | 0.3813          | 0.8726   |
-| 0.6591        | 5.0   | 10620 | 0.3372          | 0.8840   |
-| 0.6215        | 6.0   | 12744 | 0.3446          | 0.8784   |
-| 0.5979        | 7.0   | 14868 | 0.3061          | 0.8922   |
-| 0.5791        | 8.0   | 16992 | 0.3035          | 0.8904   |
-| 0.5628        | 9.0   | 19116 | 0.2915          | 0.8947   |
-| 0.5662        | 10.0  | 21240 | 0.2972          | 0.8919   |
+| 1.3829        | 1.0   | 2124  | 0.6337          | 0.8424   |
+| 1.0008        | 2.0   | 4248  | 0.4279          | 0.8789   |
+| 0.8108        | 3.0   | 6372  | 0.3942          | 0.8808   |
+| 0.7122        | 4.0   | 8496  | 0.3775          | 0.8769   |
+| 0.6473        | 5.0   | 10620 | 0.3395          | 0.8816   |
+| 0.622         | 6.0   | 12744 | 0.3346          | 0.8806   |
+| 0.5915        | 7.0   | 14868 | 0.3135          | 0.8842   |
+| 0.5661        | 8.0   | 16992 | 0.3008          | 0.8887   |
+| 0.559         | 9.0   | 19116 | 0.2914          | 0.8910   |
+| 0.5623        | 10.0  | 21240 | 0.2965          | 0.8886   |
 
 
 ### Framework versions