Add SetFit model
Browse files- README.md +28 -28
- config_setfit.json +14 -1
- model_head.pkl +2 -2
README.md
CHANGED
|
@@ -46,20 +46,20 @@ The model has been trained using an efficient few-shot learning technique that i
|
|
| 46 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
| 47 |
|
| 48 |
### Model Labels
|
| 49 |
-
| Label
|
| 50 |
-
|
| 51 |
-
|
|
| 52 |
-
|
|
| 53 |
-
|
|
| 54 |
-
|
|
| 55 |
-
|
|
| 56 |
-
|
|
| 57 |
-
|
|
| 58 |
-
|
|
| 59 |
-
|
|
| 60 |
-
|
|
| 61 |
-
|
|
| 62 |
-
|
|
| 63 |
|
| 64 |
## Uses
|
| 65 |
|
|
@@ -113,20 +113,20 @@ preds = model("The new Marvel movie is breaking box office records.")
|
|
| 113 |
|:-------------|:----|:--------|:----|
|
| 114 |
| Word count | 8 | 11.0833 | 17 |
|
| 115 |
|
| 116 |
-
| Label
|
| 117 |
-
|
| 118 |
-
|
|
| 119 |
-
|
|
| 120 |
-
|
|
| 121 |
-
|
|
| 122 |
-
|
|
| 123 |
-
|
|
| 124 |
-
|
|
| 125 |
-
|
|
| 126 |
-
|
|
| 127 |
-
|
|
| 128 |
-
|
|
| 129 |
-
|
|
| 130 |
|
| 131 |
### Training Hyperparameters
|
| 132 |
- batch_size: (16, 16)
|
|
|
|
| 46 |
- **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
|
| 47 |
|
| 48 |
### Model Labels
|
| 49 |
+
| Label | Examples |
|
| 50 |
+
|:--------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
| 51 |
+
| Politics | <ul><li>'The mayor announced a new initiative to improve public transportation.'</li><li>'The senator is facing criticism for her stance on the recent bill.'</li><li>'The upcoming election has sparked intense debates among the candidates.'</li></ul> |
|
| 52 |
+
| Health | <ul><li>'Regular exercise and a balanced diet are key to maintaining good health.'</li><li>'The World Health Organization has issued new guidelines on COVID-19.'</li><li>'A new study reveals the benefits of meditation for mental health.'</li></ul> |
|
| 53 |
+
| Finance | <ul><li>'The stock market saw a significant drop following the announcement.'</li><li>'Investing in real estate can be a profitable venture if done correctly.'</li><li>"The company's profits have doubled since the launch of their new product."</li></ul> |
|
| 54 |
+
| Travel | <ul><li>'Visiting the Grand Canyon is a breathtaking experience.'</li><li>'The tourism industry has been severely impacted by the pandemic.'</li><li>'Backpacking through Europe is a popular choice for young travelers.'</li></ul> |
|
| 55 |
+
| Food | <ul><li>'The new restaurant in town offers a fusion of Italian and Japanese cuisine.'</li><li>'Drinking eight glasses of water a day is essential for staying hydrated.'</li><li>'Cooking classes are a fun way to learn new recipes and techniques.'</li></ul> |
|
| 56 |
+
| Education | <ul><li>'The school district is implementing a new curriculum for the upcoming year.'</li><li>'Online learning has become increasingly popular during the pandemic.'</li><li>'The university is offering scholarships for students in financial need.'</li></ul> |
|
| 57 |
+
| Environment | <ul><li>'Climate change is causing a significant rise in sea levels.'</li><li>'Recycling and composting are effective ways to reduce waste.'</li><li>'The Amazon rainforest is home to millions of unique species.'</li></ul> |
|
| 58 |
+
| Fashion | <ul><li>'The new fashion trend is all about sustainability and eco-friendly materials.'</li><li>'The annual Met Gala is a major event in the fashion world.'</li><li>'Vintage clothing has made a comeback in recent years.'</li></ul> |
|
| 59 |
+
| Science | <ul><li>"NASA's Mars Rover has made significant discoveries about the red planet."</li><li>'The Nobel Prize in Physics was awarded for breakthroughs in black hole research.'</li><li>'Genetic engineering is opening up new possibilities in medical treatment.'</li></ul> |
|
| 60 |
+
| Sports | <ul><li>'The NBA Finals are set to begin next week with the top two teams in the league.'</li><li>'Serena Williams continues to dominate the tennis world with her powerful serve.'</li><li>'The World Cup is the most prestigious tournament in international soccer.'</li></ul> |
|
| 61 |
+
| Technology | <ul><li>'Artificial intelligence is changing the way we live and work.'</li><li>'The latest iPhone has a number of exciting new features.'</li><li>'Cybersecurity is becoming increasingly important as more and more data moves online.'</li></ul> |
|
| 62 |
+
| Entertainment | <ul><li>'The new Marvel movie is breaking box office records.'</li><li>'The Grammy Awards are a celebration of the best music of the year.'</li><li>'The latest season of Game of Thrones had fans on the edge of their seats.'</li></ul> |
|
| 63 |
|
| 64 |
## Uses
|
| 65 |
|
|
|
|
| 113 |
|:-------------|:----|:--------|:----|
|
| 114 |
| Word count | 8 | 11.0833 | 17 |
|
| 115 |
|
| 116 |
+
| Label | Training Sample Count |
|
| 117 |
+
|:--------------|:----------------------|
|
| 118 |
+
| Education | 3 |
|
| 119 |
+
| Entertainment | 3 |
|
| 120 |
+
| Environment | 3 |
|
| 121 |
+
| Fashion | 3 |
|
| 122 |
+
| Finance | 3 |
|
| 123 |
+
| Food | 3 |
|
| 124 |
+
| Health | 3 |
|
| 125 |
+
| Politics | 3 |
|
| 126 |
+
| Science | 3 |
|
| 127 |
+
| Sports | 3 |
|
| 128 |
+
| Technology | 3 |
|
| 129 |
+
| Travel | 3 |
|
| 130 |
|
| 131 |
### Training Hyperparameters
|
| 132 |
- batch_size: (16, 16)
|
config_setfit.json
CHANGED
|
@@ -1,4 +1,17 @@
|
|
| 1 |
{
|
| 2 |
-
"labels":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
"normalize_embeddings": false
|
| 4 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"labels": [
|
| 3 |
+
"Education",
|
| 4 |
+
"Entertainment",
|
| 5 |
+
"Environment",
|
| 6 |
+
"Fashion",
|
| 7 |
+
"Finance",
|
| 8 |
+
"Food",
|
| 9 |
+
"Health",
|
| 10 |
+
"Politics",
|
| 11 |
+
"Science",
|
| 12 |
+
"Sports",
|
| 13 |
+
"Technology",
|
| 14 |
+
"Travel"
|
| 15 |
+
],
|
| 16 |
"normalize_embeddings": false
|
| 17 |
}
|
model_head.pkl
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6ae16cf8f25bb088110133fcf0b0b18b7c0e42a670b9046ba90cc5d642b4bba5
|
| 3 |
+
size 75287
|