Add SetFit model

README.md

@@ -2,7 +2,7 @@
 base_model: HSLU-AICOMP-LearningAgencyLab/automated-essay-scoring-setfit-finetuned
 library_name: setfit
 metrics:
-- accuracy
+- qwk
 pipeline_tag: text-classification
 tags:
 - setfit
@@ -11,6 +11,20 @@ tags:
 - generated_from_setfit_trainer
 widget: []
 inference: true
+model-index:
+- name: SetFit with HSLU-AICOMP-LearningAgencyLab/automated-essay-scoring-setfit-finetuned
+  results:
+  - task:
+      type: text-classification
+      name: Text Classification
+    dataset:
+      name: Unknown
+      type: unknown
+      split: test
+    metrics:
+    - type: qwk
+      value: 0.9421066738139909
+      name: Qwk
 ---
 
 # SetFit with HSLU-AICOMP-LearningAgencyLab/automated-essay-scoring-setfit-finetuned
@@ -40,6 +54,13 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
+## Evaluation
+
+### Metrics
+| Label   | Qwk    |
+|:--------|:-------|
+| **all** | 0.9421 |
+
 ## Uses
 
 ### Direct Use for Inference
@@ -87,32 +108,6 @@ preds = model("I loved the spiderman movie!")
 
 ## Training Details
 
-### Training Hyperparameters
-- batch_size: (2, 2)
-- num_epochs: (3, 3)
-- max_steps: -1
-- sampling_strategy: oversampling
-- body_learning_rate: (2e-05, 1e-05)
-- head_learning_rate: 0.01
-- loss: CosineSimilarityLoss
-- distance_metric: cosine_distance
-- margin: 0.25
-- end_to_end: False
-- use_amp: True
-- warmup_proportion: 0.1
-- l2_weight: 0.01
-- seed: 42
-- eval_max_steps: -1
-- load_best_model_at_end: True
-
-### Training Results
-| Epoch  | Step  | Training Loss | Validation Loss |
-|:------:|:-----:|:-------------:|:---------------:|
-| 0.0001 | 1     | 0.0001        | -               |
-| 1.0    | 17265 | 0.0525        | 0.0835          |
-| 2.0    | 34530 | 0.0342        | 0.1366          |
-| 3.0    | 51795 | 0.0084        | 0.1718          |
-
 ### Framework Versions
 - Python: 3.11.9
 - SetFit: 1.1.0