End of training

README.md

@@ -2,6 +2,11 @@
 base_model: state-spaces/mamba-130m-hf
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
+- f1
+- recall
+- precision
 model-index:
 - name: mamba-text-classification-v3
   results: []
@@ -17,6 +22,12 @@ should probably proofread and complete it, then remove this comment. -->
 # mamba-text-classification-v3
 
 This model is a fine-tuned version of [state-spaces/mamba-130m-hf](https://huggingface.co/state-spaces/mamba-130m-hf) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.3637
+- Accuracy: 0.9454
+- F1: 0.9454
+- Recall: 0.9461
+- Precision: 0.9447
 
 ## Model description
 
@@ -45,6 +56,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: cosine
 - num_epochs: 5
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy | F1     | Recall | Precision |
+|:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:------:|:---------:|
+| 0.1731        | 0.9997 | 781  | 0.1553          | 0.9425   | 0.9427 | 0.9462 | 0.9393    |
+| 0.1316        | 1.9994 | 1562 | 0.1970          | 0.9319   | 0.9294 | 0.8974 | 0.9639    |
+| 0.0224        | 2.9990 | 2343 | 0.3137          | 0.9454   | 0.9455 | 0.9479 | 0.9432    |
+| 0.0002        | 4.0    | 3125 | 0.3501          | 0.9449   | 0.9450 | 0.9470 | 0.9431    |
+| 0.0004        | 4.9984 | 3905 | 0.3637          | 0.9454   | 0.9454 | 0.9461 | 0.9447    |
+
+
 ### Framework versions
 
 - Transformers 4.41.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:84162b28637f3b4595febb7e179c5063759ee2015a69ce7550ceac75204ba368
+oid sha256:a60198e0567b57773bd095a5c283e6f0a68fbdc2d51f14d9ced5b19f4149cb1c
 size 516573800