Hugging Face's logo

travelgate
/
room_type-classifier-onnx

ONNX
distilbert
Model card Files
xet
 Community
Lmunozi commited on
Commit
7cfbebf
·
verified ·
1 Parent(s): ee6964d

Add ONNX export

Browse files
Files changed (2) hide show
  1. config.json +38 -38
  2. model.onnx +1 -1
config.json CHANGED
@@ -9,47 +9,47 @@
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "room",
13
- "1": "studio",
14
- "2": "family apartment",
15
- "3": "suite",
16
- "4": "family suite",
17
- "5": "villa",
18
- "6": "family room",
19
- "7": "mansion",
20
- "8": "maisonette",
21
- "9": "default",
22
- "10": "family studio",
23
- "11": "chalet",
24
- "12": "studio suite",
25
- "13": "apartment",
26
- "14": "loft",
27
- "15": "beach house",
28
- "16": "house",
29
- "17": "bungalow",
30
- "18": "residence"
31
  },
32
  "initializer_range": 0.02,
33
  "label2id": {
34
- "apartment": 13,
35
- "beach house": 15,
36
- "bungalow": 17,
37
- "chalet": 11,
38
- "default": 9,
39
- "family apartment": 2,
40
- "family room": 6,
41
- "family studio": 10,
42
- "family suite": 4,
43
- "house": 16,
44
- "loft": 14,
45
- "maisonette": 8,
46
- "mansion": 7,
47
- "residence": 18,
48
- "room": 0,
49
- "studio": 1,
50
- "studio suite": 12,
51
- "suite": 3,
52
- "villa": 5
53
  },
54
  "max_position_embeddings": 512,
55
  "model_type": "distilbert",
 
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "maisonette",
13
+ "1": "house",
14
+ "2": "chalet",
15
+ "3": "studio",
16
+ "4": "room",
17
+ "5": "apartment",
18
+ "6": "family suite",
19
+ "7": "family apartment",
20
+ "8": "default",
21
+ "9": "residence",
22
+ "10": "studio suite",
23
+ "11": "loft",
24
+ "12": "mansion",
25
+ "13": "bungalow",
26
+ "14": "suite",
27
+ "15": "villa",
28
+ "16": "family room",
29
+ "17": "family studio",
30
+ "18": "beach house"
31
  },
32
  "initializer_range": 0.02,
33
  "label2id": {
34
+ "apartment": 5,
35
+ "beach house": 18,
36
+ "bungalow": 13,
37
+ "chalet": 2,
38
+ "default": 8,
39
+ "family apartment": 7,
40
+ "family room": 16,
41
+ "family studio": 17,
42
+ "family suite": 6,
43
+ "house": 1,
44
+ "loft": 11,
45
+ "maisonette": 0,
46
+ "mansion": 12,
47
+ "residence": 9,
48
+ "room": 4,
49
+ "studio": 3,
50
+ "studio suite": 10,
51
+ "suite": 14,
52
+ "villa": 15
53
  },
54
  "max_position_embeddings": 512,
55
  "model_type": "distilbert",
model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:63ce6d74cb838884e329f5b1348523499705fe09c14790d4c136a09330beec06
3
  size 269458984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f83a628665fe5180878af6cc61b2271fdf6448500f000fb072a537723a1fcf73
3
  size 269458984