ychu612 commited on
Commit
e43563a
·
verified ·
1 Parent(s): 9370c85

Add BERTopic model

Browse files
Files changed (2) hide show
  1. topic_embeddings.safetensors +1 -1
  2. topics.json +79 -79
topic_embeddings.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c0055b26725424feab6ca2901de36028e6d51da86d2138dec064c4fccdb75a9b
3
  size 4696
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1f3d02d653289b62334079aa6bb3e5d3782a7e6a1a7b6f9aa044b7ffd2f352d
3
  size 4696
topics.json CHANGED
@@ -3,147 +3,147 @@
3
  "-1": [
4
  [
5
  "babi",
6
- 0.08402353355198486
7
  ],
8
  [
9
  "hospit",
10
- 0.07270198174199653
11
- ],
12
- [
13
- "day",
14
- 0.06103258243670338
15
  ],
16
  [
17
  "not",
18
- 0.05721914233694409
19
  ],
20
  [
21
- "mother",
22
- 0.0520571118532909
23
  ],
24
  [
25
  "child",
26
- 0.047355295939186456
27
  ],
28
  [
29
- "born",
30
- 0.044863369045006006
31
  ],
32
  [
33
- "said",
34
- 0.04410995051900885
35
  ],
36
  [
37
- "start",
38
- 0.0404490120360765
 
 
 
 
39
  ],
40
  [
41
  "after",
42
- 0.039985293125637575
43
  ]
44
  ],
45
  "0": [
46
  [
47
  "babi",
48
- 0.13803546717670842
49
  ],
50
  [
51
- "mother",
52
- 0.0748292910678021
53
  ],
54
  [
55
- "not",
56
- 0.07302050469814218
57
  ],
58
  [
59
  "hospit",
60
- 0.056665029979960806
61
  ],
62
  [
63
- "told",
64
- 0.05447696527938358
65
  ],
66
  [
67
- "child",
68
- 0.05156465557822525
69
  ],
70
  [
71
- "month",
72
- 0.050058280395204734
73
  ],
74
  [
75
- "check",
76
- 0.04641544374294896
77
  ],
78
  [
79
- "clinic",
80
- 0.04470718882325299
81
  ],
82
  [
83
- "doctor",
84
- 0.0430379568055859
85
  ]
86
  ],
87
  "1": [
88
  [
89
  "babi",
90
- 0.1488504402678486
91
  ],
92
  [
93
  "breath",
94
- 0.11670798031827485
95
- ],
96
- [
97
- "day",
98
- 0.09737625200109924
99
  ],
100
  [
101
  "born",
102
- 0.09728182286511214
103
- ],
104
- [
105
- "see",
106
- 0.09489650404992106
107
  ],
108
  [
109
- "not",
110
- 0.08872038015761201
111
  ],
112
  [
113
  "prematur",
114
- 0.08240980059283533
115
  ],
116
  [
117
  "month",
118
- 0.07393941390325655
119
  ],
120
  [
121
  "die",
122
- 0.06822353109606293
123
  ],
124
  [
125
  "could",
126
- 0.06416885333440846
 
 
 
 
 
 
 
 
127
  ]
128
  ]
129
  },
130
  "topics": [
131
  1,
132
- -1,
133
  1,
134
  -1,
135
  -1,
136
  -1,
137
- 0,
138
  1,
139
- 0,
140
  1,
141
  -1,
142
  0,
143
  -1,
144
- -1,
145
  1,
146
- -1,
147
  0,
148
  -1,
149
  -1,
@@ -156,20 +156,20 @@
156
  0,
157
  0,
158
  -1,
159
- -1,
160
  0,
161
- 1,
162
  -1,
163
  -1,
164
  -1,
165
- 0,
 
 
166
  0,
167
  0,
168
  -1,
169
- 0,
170
  -1,
171
  -1,
172
- 0,
 
173
  0,
174
  0,
175
  0,
@@ -178,24 +178,24 @@
178
  -1,
179
  -1,
180
  -1,
181
- 0,
182
  -1,
183
  0,
184
  0,
185
  0,
186
  0,
187
- -1,
188
  -1,
189
  0,
190
  1,
191
  0,
192
  0,
193
- 0,
194
- 0,
195
  -1,
196
- 0,
197
  1,
 
198
  1,
 
199
  -1,
200
  0,
201
  -1,
@@ -203,39 +203,39 @@
203
  -1,
204
  -1,
205
  -1,
 
206
  -1,
207
  -1,
208
- 0,
209
  -1,
210
  -1,
211
  -1,
212
  1,
 
 
 
 
 
213
  0,
214
  0,
215
  -1,
216
- 0,
217
  -1,
218
  0,
219
  -1,
220
  0,
221
  -1,
222
- 0,
223
  -1,
224
- 0,
225
- 0,
226
  -1,
227
  -1,
228
  -1,
229
  0,
230
- -1,
231
  1,
232
  -1,
233
  0
234
  ],
235
  "topic_sizes": {
236
- "1": 13,
237
- "-1": 52,
238
- "0": 38
239
  },
240
  "topic_mapper": [
241
  [
@@ -258,9 +258,9 @@
258
  ]
259
  ],
260
  "topic_labels": {
261
- "-1": "-1_babi_hospit_day_not",
262
- "0": "0_babi_mother_not_hospit",
263
- "1": "1_babi_breath_day_born"
264
  },
265
  "custom_labels": null,
266
  "_outliers": 1,
 
3
  "-1": [
4
  [
5
  "babi",
6
+ 0.08710388813955308
7
  ],
8
  [
9
  "hospit",
10
+ 0.06938575099587038
 
 
 
 
11
  ],
12
  [
13
  "not",
14
+ 0.05768571799721568
15
  ],
16
  [
17
+ "day",
18
+ 0.05619094355537832
19
  ],
20
  [
21
  "child",
22
+ 0.053170858598384795
23
  ],
24
  [
25
+ "mother",
26
+ 0.052685929425289234
27
  ],
28
  [
29
+ "start",
30
+ 0.04762002802031167
31
  ],
32
  [
33
+ "doctor",
34
+ 0.04724786250296552
35
+ ],
36
+ [
37
+ "told",
38
+ 0.04651899827831358
39
  ],
40
  [
41
  "after",
42
+ 0.04561480149990497
43
  ]
44
  ],
45
  "0": [
46
  [
47
  "babi",
48
+ 0.14500885630091237
49
  ],
50
  [
51
+ "not",
52
+ 0.08149739721885443
53
  ],
54
  [
55
+ "mother",
56
+ 0.0760610489454614
57
  ],
58
  [
59
  "hospit",
60
+ 0.05815161255659558
61
  ],
62
  [
63
+ "went",
64
+ 0.04680253323244629
65
  ],
66
  [
67
+ "said",
68
+ 0.04320233836841196
69
  ],
70
  [
71
+ "check",
72
+ 0.04129769153800057
73
  ],
74
  [
75
+ "day",
76
+ 0.04093409111898061
77
  ],
78
  [
79
+ "birth",
80
+ 0.04069660867788518
81
  ],
82
  [
83
+ "told",
84
+ 0.0403533076143582
85
  ]
86
  ],
87
  "1": [
88
  [
89
  "babi",
90
+ 0.14068664009809725
91
  ],
92
  [
93
  "breath",
94
+ 0.12361997484538015
 
 
 
 
95
  ],
96
  [
97
  "born",
98
+ 0.11950584965803107
 
 
 
 
99
  ],
100
  [
101
+ "day",
102
+ 0.10367227097470466
103
  ],
104
  [
105
  "prematur",
106
+ 0.09400514941996793
107
  ],
108
  [
109
  "month",
110
+ 0.08434294952783536
111
  ],
112
  [
113
  "die",
114
+ 0.06809496916560552
115
  ],
116
  [
117
  "could",
118
+ 0.06274083147269513
119
+ ],
120
+ [
121
+ "incub",
122
+ 0.0608936619540564
123
+ ],
124
+ [
125
+ "hospit",
126
+ 0.060865076171942825
127
  ]
128
  ]
129
  },
130
  "topics": [
131
  1,
132
+ 0,
133
  1,
134
  -1,
135
  -1,
136
  -1,
137
+ -1,
138
  1,
139
+ -1,
140
  1,
141
  -1,
142
  0,
143
  -1,
144
+ 0,
145
  1,
146
+ 0,
147
  0,
148
  -1,
149
  -1,
 
156
  0,
157
  0,
158
  -1,
 
159
  0,
 
160
  -1,
161
  -1,
162
  -1,
163
+ -1,
164
+ -1,
165
+ -1,
166
  0,
167
  0,
168
  -1,
 
169
  -1,
170
  -1,
171
+ -1,
172
+ -1,
173
  0,
174
  0,
175
  0,
 
178
  -1,
179
  -1,
180
  -1,
181
+ -1,
182
  -1,
183
  0,
184
  0,
185
  0,
186
  0,
187
+ 0,
188
  -1,
189
  0,
190
  1,
191
  0,
192
  0,
 
 
193
  -1,
194
+ -1,
195
  1,
196
+ 0,
197
  1,
198
+ 0,
199
  -1,
200
  0,
201
  -1,
 
203
  -1,
204
  -1,
205
  -1,
206
+ 0,
207
  -1,
208
  -1,
 
209
  -1,
210
  -1,
211
  -1,
212
  1,
213
+ -1,
214
+ -1,
215
+ -1,
216
+ -1,
217
+ -1,
218
  0,
219
  0,
220
  -1,
 
221
  -1,
222
  0,
223
  -1,
224
  0,
225
  -1,
 
226
  -1,
 
 
227
  -1,
228
  -1,
229
  -1,
230
  0,
 
231
  1,
232
  -1,
233
  0
234
  ],
235
  "topic_sizes": {
236
+ "1": 12,
237
+ "0": 31,
238
+ "-1": 60
239
  },
240
  "topic_mapper": [
241
  [
 
258
  ]
259
  ],
260
  "topic_labels": {
261
+ "-1": "-1_babi_hospit_not_day",
262
+ "0": "0_babi_not_mother_hospit",
263
+ "1": "1_babi_breath_born_day"
264
  },
265
  "custom_labels": null,
266
  "_outliers": 1,