ychu612 commited on
Commit
f6d38c0
·
verified ·
1 Parent(s): 4157d00

Add BERTopic model

Browse files
Files changed (2) hide show
  1. topic_embeddings.safetensors +1 -1
  2. topics.json +77 -77
topic_embeddings.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:634b013a6220003f40e697c446fbdda70c961fc90db4053a86bec0419a87c1ca
3
  size 4696
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2faca4e963b73ea4827c68128c1de9d6a4234cf44b6635413b8fa5a181306d7e
3
  size 4696
topics.json CHANGED
@@ -1,142 +1,142 @@
1
  {
2
  "topic_representations": {
3
  "-1": [
4
- [
5
- "babi",
6
- 0.06764646599084934
7
- ],
8
  [
9
  "child",
10
- 0.06226965489133426
11
  ],
12
  [
13
- "hospit",
14
- 0.059188873096846874
15
  ],
16
  [
17
  "day",
18
- 0.05602310197126232
 
 
 
 
19
  ],
20
  [
21
  "not",
22
- 0.05504653727259465
23
  ],
24
  [
25
  "start",
26
- 0.05023336120171708
27
  ],
28
  [
29
- "pain",
30
- 0.046244235081077584
31
  ],
32
  [
33
- "said",
34
- 0.04434559816123986
35
  ],
36
  [
37
- "mother",
38
- 0.043452864511232085
39
  ],
40
  [
41
- "respond",
42
- 0.04289635208376847
43
  ]
44
  ],
45
  "0": [
46
  [
47
  "babi",
48
- 0.14650714309770288
49
  ],
50
  [
51
  "mother",
52
- 0.07521997654152889
53
  ],
54
  [
55
  "not",
56
- 0.07477257703502652
57
  ],
58
  [
59
  "hospit",
60
- 0.07401166898705287
61
  ],
62
  [
63
- "told",
64
- 0.05119659110836605
65
  ],
66
  [
67
- "went",
68
- 0.047503023635855686
69
  ],
70
  [
71
- "nurs",
72
- 0.04636246236333897
73
  ],
74
  [
75
- "after",
76
- 0.04405480224848352
77
  ],
78
  [
79
- "day",
80
- 0.042059671765830406
81
  ],
82
  [
83
- "month",
84
- 0.04089444089532706
85
  ]
86
  ],
87
  "1": [
88
- [
89
- "breath",
90
- 0.1335766968012524
91
- ],
92
  [
93
  "babi",
94
- 0.11736252640502529
95
  ],
96
  [
97
- "day",
98
- 0.11557861021829818
99
  ],
100
  [
101
  "born",
102
- 0.11324601943784848
103
  ],
104
  [
105
- "problem",
106
- 0.07515506343681011
107
  ],
108
  [
109
  "prematur",
110
- 0.0733608813120534
111
  ],
112
  [
113
- "not",
114
- 0.06844806322748052
115
  ],
116
  [
117
- "child",
118
- 0.06066430068026501
119
  ],
120
  [
121
- "month",
122
- 0.05923851866837377
123
  ],
124
  [
125
- "lung",
126
- 0.054277936572117616
 
 
 
 
127
  ]
128
  ]
129
  },
130
  "topics": [
131
  1,
132
- -1,
133
  1,
134
  0,
135
- -1,
136
  -1,
137
  0,
138
  1,
139
- 0,
140
  1,
141
  -1,
142
  0,
@@ -145,9 +145,9 @@
145
  1,
146
  0,
147
  0,
148
- 0,
149
  -1,
150
- 0,
 
151
  -1,
152
  -1,
153
  0,
@@ -159,33 +159,33 @@
159
  0,
160
  -1,
161
  1,
162
- 0,
163
  -1,
164
  -1,
165
  -1,
166
  0,
167
  0,
168
- -1,
169
  0,
170
  -1,
 
171
  -1,
172
  0,
173
  0,
174
  0,
175
  0,
176
  1,
 
177
  -1,
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,
@@ -196,36 +196,36 @@
196
  0,
197
  1,
198
  0,
199
- 1,
200
  0,
201
  -1,
202
  -1,
203
  -1,
204
  -1,
205
  -1,
206
- 0,
207
- -1,
208
  -1,
209
  -1,
210
  0,
211
  -1,
 
 
212
  1,
213
  -1,
214
  0,
215
  -1,
216
  0,
217
  -1,
218
- -1,
219
  0,
220
  0,
221
  0,
222
- -1,
223
- -1,
 
224
  0,
225
  0,
226
  -1,
227
  -1,
228
- 1,
229
  -1,
230
  0,
231
  -1,
@@ -233,9 +233,9 @@
233
  0
234
  ],
235
  "topic_sizes": {
236
- "1": 14,
237
- "-1": 44,
238
- "0": 45
239
  },
240
  "topic_mapper": [
241
  [
@@ -248,19 +248,19 @@
248
  0,
249
  0,
250
  0,
251
- 0
252
  ],
253
  [
254
  1,
255
  1,
256
  1,
257
- 1
258
  ]
259
  ],
260
  "topic_labels": {
261
- "-1": "-1_babi_child_hospit_day",
262
  "0": "0_babi_mother_not_hospit",
263
- "1": "1_breath_babi_day_born"
264
  },
265
  "custom_labels": null,
266
  "_outliers": 1,
 
1
  {
2
  "topic_representations": {
3
  "-1": [
 
 
 
 
4
  [
5
  "child",
6
+ 0.0862729115208402
7
  ],
8
  [
9
+ "babi",
10
+ 0.06792185071142329
11
  ],
12
  [
13
  "day",
14
+ 0.06701179395539683
15
+ ],
16
+ [
17
+ "hospit",
18
+ 0.06697218258034836
19
  ],
20
  [
21
  "not",
22
+ 0.06335901308553044
23
  ],
24
  [
25
  "start",
26
+ 0.05173279680669054
27
  ],
28
  [
29
+ "doctor",
30
+ 0.04790271853211265
31
  ],
32
  [
33
+ "mother",
34
+ 0.04513512794565841
35
  ],
36
  [
37
+ "said",
38
+ 0.04503718671610513
39
  ],
40
  [
41
+ "told",
42
+ 0.0448716316329543
43
  ]
44
  ],
45
  "0": [
46
  [
47
  "babi",
48
+ 0.13352230835829582
49
  ],
50
  [
51
  "mother",
52
+ 0.07244619262615236
53
  ],
54
  [
55
  "not",
56
+ 0.06760006804365859
57
  ],
58
  [
59
  "hospit",
60
+ 0.06449363815583953
61
  ],
62
  [
63
+ "nurs",
64
+ 0.047325911254081955
65
  ],
66
  [
67
+ "told",
68
+ 0.045604010970912545
69
  ],
70
  [
71
+ "birth",
72
+ 0.04381844767089936
73
  ],
74
  [
75
+ "month",
76
+ 0.042762412626851556
77
  ],
78
  [
79
+ "went",
80
+ 0.04245543715913095
81
  ],
82
  [
83
+ "clinic",
84
+ 0.04051155009932372
85
  ]
86
  ],
87
  "1": [
 
 
 
 
88
  [
89
  "babi",
90
+ 0.14547144788603442
91
  ],
92
  [
93
+ "breath",
94
+ 0.12097128553538077
95
  ],
96
  [
97
  "born",
98
+ 0.10859207343355334
99
  ],
100
  [
101
+ "day",
102
+ 0.09420448367107866
103
  ],
104
  [
105
  "prematur",
106
+ 0.08542020426745944
107
  ],
108
  [
109
+ "month",
110
+ 0.07664039705953989
111
  ],
112
  [
113
+ "pas",
114
+ 0.06340569530439655
115
  ],
116
  [
117
+ "die",
118
+ 0.06187625052034473
119
  ],
120
  [
121
+ "hospit",
122
+ 0.06145180191890726
123
+ ],
124
+ [
125
+ "not",
126
+ 0.06130753819110479
127
  ]
128
  ]
129
  },
130
  "topics": [
131
  1,
132
+ 0,
133
  1,
134
  0,
135
+ 0,
136
  -1,
137
  0,
138
  1,
139
+ -1,
140
  1,
141
  -1,
142
  0,
 
145
  1,
146
  0,
147
  0,
 
148
  -1,
149
+ -1,
150
+ -1,
151
  -1,
152
  -1,
153
  0,
 
159
  0,
160
  -1,
161
  1,
 
162
  -1,
163
  -1,
164
  -1,
165
  0,
166
  0,
167
+ 0,
168
  0,
169
  -1,
170
+ 0,
171
  -1,
172
  0,
173
  0,
174
  0,
175
  0,
176
  1,
177
+ 0,
178
  -1,
179
  -1,
180
  -1,
181
+ 0,
 
182
  -1,
183
  0,
184
  0,
185
  0,
186
+ -1,
187
  0,
188
  0,
 
189
  0,
190
  1,
191
  0,
 
196
  0,
197
  1,
198
  0,
199
+ -1,
200
  0,
201
  -1,
202
  -1,
203
  -1,
204
  -1,
205
  -1,
 
 
206
  -1,
207
  -1,
208
  0,
209
  -1,
210
+ -1,
211
+ 1,
212
  1,
213
  -1,
214
  0,
215
  -1,
216
  0,
217
  -1,
 
218
  0,
219
  0,
220
  0,
221
+ 0,
222
+ 0,
223
+ 0,
224
  0,
225
  0,
226
  -1,
227
  -1,
228
+ -1,
229
  -1,
230
  0,
231
  -1,
 
233
  0
234
  ],
235
  "topic_sizes": {
236
+ "1": 13,
237
+ "0": 49,
238
+ "-1": 41
239
  },
240
  "topic_mapper": [
241
  [
 
248
  0,
249
  0,
250
  0,
251
+ 1
252
  ],
253
  [
254
  1,
255
  1,
256
  1,
257
+ 0
258
  ]
259
  ],
260
  "topic_labels": {
261
+ "-1": "-1_child_babi_day_hospit",
262
  "0": "0_babi_mother_not_hospit",
263
+ "1": "1_babi_breath_born_day"
264
  },
265
  "custom_labels": null,
266
  "_outliers": 1,