Add SetFit model

README.md

@@ -281,13 +281,13 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.625
+      value: 0.4583333333333333
       name: Accuracy
 ---
 
 # SetFit
 
-This is a [SetFit](https://github.com/huggingface/setfit) model trained on the [CabraVC/vector_dataset_stratified_ttv_split_2023-12-05_21-07](https://huggingface.co/datasets/CabraVC/vector_dataset_stratified_ttv_split_2023-12-05_21-07) dataset that can be used for Text Classification. A MLP instance is used for classification.
+This is a [SetFit](https://github.com/huggingface/setfit) model trained on the [CabraVC/vector_dataset_stratified_ttv_split_2023-12-05_21-07](https://huggingface.co/datasets/CabraVC/vector_dataset_stratified_ttv_split_2023-12-05_21-07) dataset that can be used for Text Classification. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
 
 The model has been trained using an efficient few-shot learning technique that involves:
 
@@ -299,7 +299,7 @@ The model has been trained using an efficient few-shot learning technique that i
 ### Model Description
 - **Model Type:** SetFit
 <!-- - **Sentence Transformer:** [Unknown](https://huggingface.co/unknown) -->
-- **Classification head:** a MLP instance
+- **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
 - **Maximum Sequence Length:** 512 tokens
 - **Number of Classes:** 3 classes
 - **Training Dataset:** [CabraVC/vector_dataset_stratified_ttv_split_2023-12-05_21-07](https://huggingface.co/datasets/CabraVC/vector_dataset_stratified_ttv_split_2023-12-05_21-07)
@@ -324,7 +324,7 @@ The model has been trained using an efficient few-shot learning technique that i
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.625    |
+| **all** | 0.4583   |
 
 ## Uses
 

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:53c017b87f9d139abbbcae9bc86ff1cad970e1c3eea7b5dfbbe766f72b77bbc8
-size 799300
+oid sha256:3e00150cf01ce43dc8fc03354e31b4380f0d111dc53d782ffa5a646a7b721f09
+size 19471