Add SetFit model

README.md

@@ -9,10 +9,11 @@ base_model: BAAI/bge-small-en-v1.5
 metrics:
 - accuracy
 widget:
-- text: acquirer growth affects m & a success
-- text: board size affects ceo pay
-- text: market orientation affects entrepreneurship intention
-- text: option sales taxes and fiscal affects ceo pay
+- text: variables affects cost of debt
+- text: variables affects cost of debt
+- text: personal attitude affects entrepreneurship intention
+- text: positive emotionality affects enterprise systems success
+- text: video affects firm performance
 pipeline_tag: text-classification
 inference: true
 ---
@@ -45,10 +46,10 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
 ### Model Labels
-| Label | Examples                                                                                                                          |
-|:------|:----------------------------------------------------------------------------------------------------------------------------------|
-| 0     | <ul><li>'option sales taxes and fiscal affects ceo pay'</li><li>'market orientation affects entrepreneurship intention'</li></ul> |
-| 1     | <ul><li>'board size affects ceo pay'</li><li>'acquirer growth affects m & a success'</li></ul>                                    |
+| Label | Examples                                                                                                                                                                   |
+|:------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| 1     | <ul><li>'ease of use affects enterprise systems success'</li><li>'acquirer growth affects m & a success'</li><li>'high yield affects cost of debt'</li></ul>               |
+| 0     | <ul><li>'option sales taxes and fiscal affects ceo pay'</li><li>'variables affects cost of debt'</li><li>'market orientation affects entrepreneurship intention'</li></ul> |
 
 ## Uses
 
@@ -68,7 +69,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("abehandler/setfit")
 # Run inference
-preds = model("board size affects ceo pay")
+preds = model("variables affects cost of debt")
 ```
 
 <!--
@@ -100,12 +101,12 @@ preds = model("board size affects ceo pay")
 ### Training Set Metrics
 | Training set | Min | Median | Max |
 |:-------------|:----|:-------|:----|
-| Word count   | 5   | 6.25   | 8   |
+| Word count   | 4   | 5.5    | 8   |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 2                     |
-| 1     | 2                     |
+| 0     | 7                     |
+| 1     | 9                     |
 
 ### Training Hyperparameters
 - batch_size: (32, 32)
@@ -127,7 +128,8 @@ preds = model("board size affects ceo pay")
 ### Training Results
 | Epoch | Step | Training Loss | Validation Loss |
 |:-----:|:----:|:-------------:|:---------------:|
-| 1.0   | 1    | 0.2472        | -               |
+| 0.2   | 1    | 0.2372        | -               |
+| 10.0  | 50   | 0.1078        | -               |
 
 ### Framework Versions
 - Python: 3.10.12

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4b31920f83b1159456d05060f05d2519b93dae07f377d6abc714129153c5cb91
+oid sha256:daa3dd0b9c1cf8067cc8f93083ca7cd9835014a42bc3fd1b611993a2c01c765f
 size 133462128

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63528f69c4997cfb2771bc1f3c48803e7122ba7f4ac91965244fcf33f9ff0e1f
+oid sha256:1eb400717286ef93943eaa198b1c83035c14ebee8811b10561ab7c05d07babca
 size 3935