Hugging Face's logo

mitra-mir
/
setfit-model-Ireland_4labels_unbalanced_data_3epochs

Sentence Similarity
sentence-transformers
PyTorch
mpnet
feature-extraction
Inference Endpoints
Model card Files Community
mitra-mir commited on
Commit
aecc116
•
1 Parent(s): 80bb6c9

Add SetFit model

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. model_head.pkl +1 -1
  3. pytorch_model.bin +1 -1
README.md CHANGED
@@ -58,7 +58,7 @@ The model was trained with the parameters:
58
  Parameters of the fit()-Method:
59
  ```
60
  {
61
- "epochs": 3,
62
  "evaluation_steps": 0,
63
  "evaluator": "NoneType",
64
  "max_grad_norm": 1,
@@ -67,8 +67,8 @@ Parameters of the fit()-Method:
67
  "lr": 2e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
- "steps_per_epoch": 444,
71
- "warmup_steps": 45,
72
  "weight_decay": 0.01
73
  }
74
  ```
 
58
  Parameters of the fit()-Method:
59
  ```
60
  {
61
+ "epochs": 4,
62
  "evaluation_steps": 0,
63
  "evaluator": "NoneType",
64
  "max_grad_norm": 1,
 
67
  "lr": 2e-05
68
  },
69
  "scheduler": "WarmupLinear",
70
+ "steps_per_epoch": 592,
71
+ "warmup_steps": 60,
72
  "weight_decay": 0.01
73
  }
74
  ```
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:316c654e8557e98778eeafb8238132b0b8e73c96eb87fa9c5cb0004f2e18873e
3
  size 25463
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:206a0533c4d6bdc52cc461bd6669a25a97617d2da78a460d7c3f891c41b980fb
3
  size 25463
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9285df3ba4cae799d091185865491f7ca91d466e2643ffc512ca537f0b8a7fe9
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcc188baf518b3938811dd6c17a602216f1d60dedf98dbb54c7f597f2d94dc9e
3
  size 438016493