Upload files with new model

#2
by mansonpr13 - opened
onnx/config.json CHANGED
@@ -1,44 +1,27 @@
1
  {
2
  "architectures": [
3
- "DebertaV2ForSequenceClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
 
6
  "dtype": "float32",
 
7
  "hidden_act": "gelu",
8
  "hidden_dropout_prob": 0.1,
9
  "hidden_size": 768,
10
- "id2label": {
11
- "0": "no-claim",
12
- "1": "claim"
13
- },
14
  "initializer_range": 0.02,
15
- "label2id": {
16
- "no-claim": 0,
17
- "claim": 1
18
- },
19
  "intermediate_size": 3072,
20
- "layer_norm_eps": 1e-07,
21
- "legacy": true,
22
- "max_position_embeddings": 512,
23
- "max_relative_positions": -1,
24
- "model_type": "deberta-v2",
25
- "norm_rel_ebd": "layer_norm",
26
  "num_attention_heads": 12,
27
  "num_hidden_layers": 12,
28
- "num_labels": 2,
29
- "pad_token_id": 0,
30
- "pooler_dropout": 0,
31
- "pooler_hidden_act": "gelu",
32
- "pooler_hidden_size": 768,
33
- "pos_att_type": [
34
- "p2c",
35
- "c2p"
36
- ],
37
- "position_biased_input": false,
38
- "position_buckets": 256,
39
- "relative_attention": true,
40
- "share_att_key": true,
41
  "transformers_version": "4.57.6",
42
- "type_vocab_size": 0,
43
- "vocab_size": 251000
 
44
  }
 
1
  {
2
  "architectures": [
3
+ "XLMRobertaForSequenceClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": null,
8
  "dtype": "float32",
9
+ "eos_token_id": 2,
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
 
 
 
 
13
  "initializer_range": 0.02,
 
 
 
 
14
  "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "xlm-roberta",
 
 
 
18
  "num_attention_heads": 12,
19
  "num_hidden_layers": 12,
20
+ "output_past": true,
21
+ "pad_token_id": 1,
22
+ "position_embedding_type": "absolute",
 
 
 
 
 
 
 
 
 
 
23
  "transformers_version": "4.57.6",
24
+ "type_vocab_size": 1,
25
+ "use_cache": true,
26
+ "vocab_size": 250002
27
  }
onnx/golden_vectors.json CHANGED
@@ -1,67 +1,67 @@
1
  {
2
- "threshold": 0.5,
3
- "max_length": 64,
4
  "atol": 0.05,
5
  "cases": [
6
  {
7
  "text": "El PIB de España creció un 3% en 2023.",
8
  "expected_is_claim": true,
9
- "prob_claim": 0.9986067306257296
10
  },
11
  {
12
  "text": "La Torre Eiffel mide 330 metros de altura.",
13
  "expected_is_claim": true,
14
- "prob_claim": 0.9985684722293962
15
  },
16
  {
17
  "text": "The company was founded in 1998 in California.",
18
  "expected_is_claim": true,
19
- "prob_claim": 0.9983154952973389
20
  },
21
  {
22
  "text": "Water boils at 100 degrees Celsius at sea level.",
23
  "expected_is_claim": true,
24
- "prob_claim": 0.9985683713448339
25
  },
26
  {
27
  "text": "Le mont Blanc est le plus haut sommet des Alpes.",
28
  "expected_is_claim": true,
29
- "prob_claim": 0.9984872762823668
30
  },
31
  {
32
  "text": "Il fatturato è aumentato del 12% nel primo trimestre.",
33
  "expected_is_claim": true,
34
- "prob_claim": 0.9984786113378151
35
  },
36
  {
37
  "text": "¡Buena pregunta!",
38
  "expected_is_claim": false,
39
- "prob_claim": 0.0028797696267300753
40
  },
41
  {
42
  "text": "Espero que esto te ayude.",
43
  "expected_is_claim": false,
44
- "prob_claim": 0.003178733618203916
45
  },
46
  {
47
  "text": "Great, let me know if you need anything else.",
48
  "expected_is_claim": false,
49
- "prob_claim": 0.002844390951936898
50
  },
51
  {
52
  "text": "Claro, con gusto te explico.",
53
  "expected_is_claim": false,
54
- "prob_claim": 0.002913908149180559
55
  },
56
  {
57
  "text": "Thanks for reaching out!",
58
  "expected_is_claim": false,
59
- "prob_claim": 0.0035664084808701845
60
  },
61
  {
62
  "text": "En resumen, aquí tienes la información.",
63
  "expected_is_claim": false,
64
- "prob_claim": 0.0028530380945179746
65
  }
66
  ]
67
  }
 
1
  {
2
+ "threshold": 0.03,
3
+ "max_length": 80,
4
  "atol": 0.05,
5
  "cases": [
6
  {
7
  "text": "El PIB de España creció un 3% en 2023.",
8
  "expected_is_claim": true,
9
+ "prob_claim": 0.9999027114474605
10
  },
11
  {
12
  "text": "La Torre Eiffel mide 330 metros de altura.",
13
  "expected_is_claim": true,
14
+ "prob_claim": 0.9999603669293664
15
  },
16
  {
17
  "text": "The company was founded in 1998 in California.",
18
  "expected_is_claim": true,
19
+ "prob_claim": 0.0018300915078948478
20
  },
21
  {
22
  "text": "Water boils at 100 degrees Celsius at sea level.",
23
  "expected_is_claim": true,
24
+ "prob_claim": 0.9999487235506652
25
  },
26
  {
27
  "text": "Le mont Blanc est le plus haut sommet des Alpes.",
28
  "expected_is_claim": true,
29
+ "prob_claim": 0.9997233576112876
30
  },
31
  {
32
  "text": "Il fatturato è aumentato del 12% nel primo trimestre.",
33
  "expected_is_claim": true,
34
+ "prob_claim": 0.9999609237534306
35
  },
36
  {
37
  "text": "¡Buena pregunta!",
38
  "expected_is_claim": false,
39
+ "prob_claim": 0.00018462060029356833
40
  },
41
  {
42
  "text": "Espero que esto te ayude.",
43
  "expected_is_claim": false,
44
+ "prob_claim": 0.0001773333486500023
45
  },
46
  {
47
  "text": "Great, let me know if you need anything else.",
48
  "expected_is_claim": false,
49
+ "prob_claim": 0.00016158439134309846
50
  },
51
  {
52
  "text": "Claro, con gusto te explico.",
53
  "expected_is_claim": false,
54
+ "prob_claim": 0.00017395973177061248
55
  },
56
  {
57
  "text": "Thanks for reaching out!",
58
  "expected_is_claim": false,
59
+ "prob_claim": 0.00018592228379539863
60
  },
61
  {
62
  "text": "En resumen, aquí tienes la información.",
63
  "expected_is_claim": false,
64
+ "prob_claim": 0.00019974465462584796
65
  }
66
  ]
67
  }
onnx/model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:05aa659d235fb84b0f92f193441da49d70cde88009a610419b5dc9cafa10fc1c
3
- size 1116153796
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed2bac8e56c40cef1a7a1d70173bc768a23c8c0b0f0c2d570d2e619a04667545
3
+ size 1112403710
onnx/model_quantized.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b80fe4f83e983b30ff6bb9f1a1d60dc5d4b3f9d5edcefaba610d56502966fc84
3
- size 561381900
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2bf63b76d04edc74ce0fdbb6dbc58e555940450d3dc67f3dfb27714417cab830
3
+ size 278822345
onnx/threshold.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "threshold": 0.5,
3
- "model_key": "mdeberta",
4
- "model_name": "microsoft/mdeberta-v3-base",
5
- "max_length": 64,
6
  "labels": {
7
  "0": "no-claim",
8
  "1": "claim"
 
1
  {
2
+ "threshold": 0.03,
3
+ "model_key": "xlmr",
4
+ "model_name": "FacebookAI/xlm-roberta-base",
5
+ "max_length": 80,
6
  "labels": {
7
  "0": "no-claim",
8
  "1": "claim"
onnx/tokenizer.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:35e2b7254bee739517e422891a9e0bfdd8a717b1826c8965e52a94e92be12544
3
- size 16350861
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44f100dd94b78365d1179371e6b2e0e816806738e05a71bd459cc401a2152b83
3
+ size 17082831