Hugging Face's logo

AswanthCManoj
/
filler-classifier-e5-small

Model card Files
xet
 Community
filler-classifier-e5-small
File size: 1,171 Bytes
52345b2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
{
  "model_name": "intfloat/multilingual-e5-small",
  "pooling": "mean",
  "text_prefix": "query: ",
  "max_len": 128,
  "function_labels": [
    "planning",
    "receipt",
    "appreciation",
    "affiliation",
    "deliberation",
    "brisk",
    "transition"
  ],
  "tone_labels": [
    "thoughtful",
    "attentive",
    "gentle",
    "pleased",
    "brisk",
    "reflective"
  ],
  "tone_collapse": {
    "curious": "thoughtful",
    "calm": "attentive",
    "focused": "brisk",
    "concerned": "gentle"
  },
  "num_functions": 7,
  "num_tones": 6,
  "dropout": 0.15,
  "contrastive_lambda": 0.3,
  "training_info": {
    "lr": 3e-05,
    "epochs_run": 7,
    "best_val_loss": 1.5853182556806669,
    "test_func_acc": 0.7118353344768439,
    "test_tone_acc": 0.6826758147512865,
    "cpu_latency_ms_colab": 80.43561199997384,
    "params": 117658765,
    "train_rows": 9965,
    "val_rows": 1303,
    "test_rows": 1166,
    "label_files": [
      "spring_ml.jsonl",
      "synthetic_targeted_v1.jsonl",
      "synthetic_targeted_v2.jsonl",
      "synthetic_targeted_v3.jsonl",
      "synthetic_targeted_v4.jsonl"
    ],
    "variant": "contrastive_candidates"
  }
}