AngelPanizo commited on
Commit
76d718f
·
verified ·
1 Parent(s): 06ac83f

Add BERTopic model

Browse files
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - bertopic
5
+ library_name: bertopic
6
+ pipeline_tag: text-classification
7
+ ---
8
+
9
+ # MARTINI_enrich_BERTopic_truthwins888
10
+
11
+ This is a [BERTopic](https://github.com/MaartenGr/BERTopic) model.
12
+ BERTopic is a flexible and modular topic modeling framework that allows for the generation of easily interpretable topics from large datasets.
13
+
14
+ ## Usage
15
+
16
+ To use this model, please install BERTopic:
17
+
18
+ ```
19
+ pip install -U bertopic
20
+ ```
21
+
22
+ You can use the model as follows:
23
+
24
+ ```python
25
+ from bertopic import BERTopic
26
+ topic_model = BERTopic.load("AIDA-UPM/MARTINI_enrich_BERTopic_truthwins888")
27
+
28
+ topic_model.get_topic_info()
29
+ ```
30
+
31
+ ## Topic overview
32
+
33
+ * Number of topics: 5
34
+ * Number of training documents: 459
35
+
36
+ <details>
37
+ <summary>Click here for an overview of all topics.</summary>
38
+
39
+ | Topic ID | Topic Keywords | Topic Frequency | Label |
40
+ |----------|----------------|-----------------|-------|
41
+ | -1 | protests - tyranny - fauci - everyone - sweden | 23 | -1_protests_tyranny_fauci_everyone |
42
+ | 0 | vaccinated - fauci - deaths - 2021 - myocarditis | 264 | 0_vaccinated_fauci_deaths_2021 |
43
+ | 1 | guadeloupe - protests - toulon - genova - green | 109 | 1_guadeloupe_protests_toulon_genova |
44
+ | 2 | austria - compulsory - vaccinated - restrictions - nationalists | 34 | 2_austria_compulsory_vaccinated_restrictions |
45
+ | 3 | australia - tyranny - protest - leader - umberto | 29 | 3_australia_tyranny_protest_leader |
46
+
47
+ </details>
48
+
49
+ ## Training hyperparameters
50
+
51
+ * calculate_probabilities: True
52
+ * language: None
53
+ * low_memory: False
54
+ * min_topic_size: 10
55
+ * n_gram_range: (1, 1)
56
+ * nr_topics: None
57
+ * seed_topic_list: None
58
+ * top_n_words: 10
59
+ * verbose: False
60
+ * zeroshot_min_similarity: 0.7
61
+ * zeroshot_topic_list: None
62
+
63
+ ## Framework versions
64
+
65
+ * Numpy: 1.26.4
66
+ * HDBSCAN: 0.8.40
67
+ * UMAP: 0.5.7
68
+ * Pandas: 2.2.3
69
+ * Scikit-Learn: 1.5.2
70
+ * Sentence-transformers: 3.3.1
71
+ * Transformers: 4.46.3
72
+ * Numba: 0.60.0
73
+ * Plotly: 5.24.1
74
+ * Python: 3.10.12
config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "calculate_probabilities": true,
3
+ "language": null,
4
+ "low_memory": false,
5
+ "min_topic_size": 10,
6
+ "n_gram_range": [
7
+ 1,
8
+ 1
9
+ ],
10
+ "nr_topics": null,
11
+ "seed_topic_list": null,
12
+ "top_n_words": 10,
13
+ "verbose": false,
14
+ "zeroshot_min_similarity": 0.7,
15
+ "zeroshot_topic_list": null
16
+ }
ctfidf.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44e367b60494e177a1e81596c8cfa8607e42bba3f40e175c8bef15c4313bd6e2
3
+ size 105756
ctfidf_config.json ADDED
The diff for this file is too large to render. See raw diff
 
topic_embeddings.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe8a69c32d69d6c71af37b50a3732eac7ffe1a82c5efd9b882b2f937fb8ff437
3
+ size 20568
topics.json ADDED
@@ -0,0 +1,619 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "topic_representations": {
3
+ "-1": [
4
+ [
5
+ "protests",
6
+ 0.590228796005249
7
+ ],
8
+ [
9
+ "tyranny",
10
+ 0.526910662651062
11
+ ],
12
+ [
13
+ "fauci",
14
+ 0.5227830410003662
15
+ ],
16
+ [
17
+ "everyone",
18
+ 0.49442145228385925
19
+ ],
20
+ [
21
+ "sweden",
22
+ 0.48835963010787964
23
+ ]
24
+ ],
25
+ "0": [
26
+ [
27
+ "vaccinated",
28
+ 0.6199644804000854
29
+ ],
30
+ [
31
+ "fauci",
32
+ 0.5790196657180786
33
+ ],
34
+ [
35
+ "deaths",
36
+ 0.5278588533401489
37
+ ],
38
+ [
39
+ "2021",
40
+ 0.4666038155555725
41
+ ],
42
+ [
43
+ "myocarditis",
44
+ 0.4568958282470703
45
+ ]
46
+ ],
47
+ "1": [
48
+ [
49
+ "guadeloupe",
50
+ 0.6012896299362183
51
+ ],
52
+ [
53
+ "protests",
54
+ 0.5979149341583252
55
+ ],
56
+ [
57
+ "toulon",
58
+ 0.54925537109375
59
+ ],
60
+ [
61
+ "genova",
62
+ 0.5330566763877869
63
+ ],
64
+ [
65
+ "green",
66
+ 0.5111173391342163
67
+ ]
68
+ ],
69
+ "2": [
70
+ [
71
+ "austria",
72
+ 0.5842912197113037
73
+ ],
74
+ [
75
+ "compulsory",
76
+ 0.5449857115745544
77
+ ],
78
+ [
79
+ "vaccinated",
80
+ 0.5443344116210938
81
+ ],
82
+ [
83
+ "restrictions",
84
+ 0.4983295202255249
85
+ ],
86
+ [
87
+ "nationalists",
88
+ 0.49358806014060974
89
+ ]
90
+ ],
91
+ "3": [
92
+ [
93
+ "australia",
94
+ 0.6613942384719849
95
+ ],
96
+ [
97
+ "tyranny",
98
+ 0.5437164306640625
99
+ ],
100
+ [
101
+ "protest",
102
+ 0.5156422853469849
103
+ ],
104
+ [
105
+ "leader",
106
+ 0.492281973361969
107
+ ],
108
+ [
109
+ "umberto",
110
+ 0.477047860622406
111
+ ]
112
+ ]
113
+ },
114
+ "topics": [
115
+ 0,
116
+ 0,
117
+ -1,
118
+ -1,
119
+ 2,
120
+ -1,
121
+ -1,
122
+ -1,
123
+ -1,
124
+ 1,
125
+ -1,
126
+ 0,
127
+ -1,
128
+ -1,
129
+ -1,
130
+ 0,
131
+ 0,
132
+ -1,
133
+ -1,
134
+ -1,
135
+ -1,
136
+ -1,
137
+ -1,
138
+ -1,
139
+ -1,
140
+ -1,
141
+ 0,
142
+ -1,
143
+ -1,
144
+ -1,
145
+ -1,
146
+ -1,
147
+ -1,
148
+ 0,
149
+ -1,
150
+ -1,
151
+ 0,
152
+ -1,
153
+ -1,
154
+ 0,
155
+ 0,
156
+ -1,
157
+ -1,
158
+ 3,
159
+ 1,
160
+ 1,
161
+ 0,
162
+ 3,
163
+ -1,
164
+ -1,
165
+ -1,
166
+ 0,
167
+ -1,
168
+ -1,
169
+ -1,
170
+ -1,
171
+ -1,
172
+ 0,
173
+ -1,
174
+ -1,
175
+ 1,
176
+ -1,
177
+ -1,
178
+ 0,
179
+ 2,
180
+ -1,
181
+ 1,
182
+ -1,
183
+ -1,
184
+ -1,
185
+ -1,
186
+ -1,
187
+ -1,
188
+ 0,
189
+ 2,
190
+ 2,
191
+ -1,
192
+ -1,
193
+ -1,
194
+ -1,
195
+ -1,
196
+ 2,
197
+ -1,
198
+ 1,
199
+ 1,
200
+ -1,
201
+ 3,
202
+ -1,
203
+ 0,
204
+ -1,
205
+ 2,
206
+ -1,
207
+ 0,
208
+ -1,
209
+ -1,
210
+ -1,
211
+ -1,
212
+ 0,
213
+ 0,
214
+ -1,
215
+ -1,
216
+ 0,
217
+ 0,
218
+ -1,
219
+ -1,
220
+ 0,
221
+ -1,
222
+ 0,
223
+ -1,
224
+ -1,
225
+ -1,
226
+ -1,
227
+ -1,
228
+ 0,
229
+ -1,
230
+ -1,
231
+ -1,
232
+ -1,
233
+ -1,
234
+ -1,
235
+ -1,
236
+ 0,
237
+ 0,
238
+ -1,
239
+ -1,
240
+ 2,
241
+ -1,
242
+ 0,
243
+ -1,
244
+ 0,
245
+ -1,
246
+ 0,
247
+ -1,
248
+ -1,
249
+ -1,
250
+ -1,
251
+ 0,
252
+ -1,
253
+ 2,
254
+ -1,
255
+ 2,
256
+ 1,
257
+ -1,
258
+ -1,
259
+ 2,
260
+ 3,
261
+ 3,
262
+ 0,
263
+ 0,
264
+ 0,
265
+ -1,
266
+ 2,
267
+ 0,
268
+ 3,
269
+ 0,
270
+ -1,
271
+ -1,
272
+ -1,
273
+ -1,
274
+ 0,
275
+ 0,
276
+ 0,
277
+ -1,
278
+ -1,
279
+ -1,
280
+ -1,
281
+ 0,
282
+ 3,
283
+ -1,
284
+ -1,
285
+ -1,
286
+ -1,
287
+ -1,
288
+ -1,
289
+ -1,
290
+ -1,
291
+ -1,
292
+ -1,
293
+ 0,
294
+ -1,
295
+ 3,
296
+ 3,
297
+ 0,
298
+ 2,
299
+ 1,
300
+ 2,
301
+ 2,
302
+ 1,
303
+ 0,
304
+ 1,
305
+ -1,
306
+ 0,
307
+ -1,
308
+ -1,
309
+ -1,
310
+ -1,
311
+ 0,
312
+ -1,
313
+ 0,
314
+ 0,
315
+ 3,
316
+ -1,
317
+ 2,
318
+ 3,
319
+ -1,
320
+ 0,
321
+ 1,
322
+ 0,
323
+ 3,
324
+ 1,
325
+ 1,
326
+ 3,
327
+ 2,
328
+ 3,
329
+ 0,
330
+ -1,
331
+ 2,
332
+ 0,
333
+ 0,
334
+ -1,
335
+ 2,
336
+ -1,
337
+ 0,
338
+ 2,
339
+ 2,
340
+ -1,
341
+ 0,
342
+ -1,
343
+ 0,
344
+ 0,
345
+ 0,
346
+ 0,
347
+ 2,
348
+ 0,
349
+ -1,
350
+ -1,
351
+ -1,
352
+ -1,
353
+ 0,
354
+ 0,
355
+ -1,
356
+ 2,
357
+ 2,
358
+ -1,
359
+ -1,
360
+ 0,
361
+ -1,
362
+ 0,
363
+ -1,
364
+ -1,
365
+ -1,
366
+ -1,
367
+ 1,
368
+ -1,
369
+ 0,
370
+ 3,
371
+ -1,
372
+ -1,
373
+ 3,
374
+ 3,
375
+ 1,
376
+ 0,
377
+ -1,
378
+ 0,
379
+ -1,
380
+ -1,
381
+ -1,
382
+ 0,
383
+ 0,
384
+ 0,
385
+ 0,
386
+ -1,
387
+ -1,
388
+ -1,
389
+ -1,
390
+ 0,
391
+ 1,
392
+ -1,
393
+ -1,
394
+ -1,
395
+ 0,
396
+ -1,
397
+ -1,
398
+ -1,
399
+ -1,
400
+ -1,
401
+ 2,
402
+ -1,
403
+ -1,
404
+ -1,
405
+ 0,
406
+ -1,
407
+ -1,
408
+ -1,
409
+ -1,
410
+ -1,
411
+ 3,
412
+ -1,
413
+ 0,
414
+ 0,
415
+ -1,
416
+ -1,
417
+ -1,
418
+ 0,
419
+ 0,
420
+ 0,
421
+ 0,
422
+ -1,
423
+ -1,
424
+ -1,
425
+ -1,
426
+ -1,
427
+ 0,
428
+ 0,
429
+ -1,
430
+ -1,
431
+ 0,
432
+ 0,
433
+ 1,
434
+ -1,
435
+ 0,
436
+ -1,
437
+ -1,
438
+ -1,
439
+ -1,
440
+ -1,
441
+ -1,
442
+ 0,
443
+ 2,
444
+ -1,
445
+ -1,
446
+ 1,
447
+ -1,
448
+ -1,
449
+ -1,
450
+ 1,
451
+ 1,
452
+ -1,
453
+ 1,
454
+ 1,
455
+ 0,
456
+ -1,
457
+ 0,
458
+ -1,
459
+ 0,
460
+ -1,
461
+ -1,
462
+ 1,
463
+ 3,
464
+ 0,
465
+ 1,
466
+ 1,
467
+ -1,
468
+ -1,
469
+ 1,
470
+ -1,
471
+ 0,
472
+ 1,
473
+ 0,
474
+ -1,
475
+ -1,
476
+ -1,
477
+ -1,
478
+ -1,
479
+ -1,
480
+ 1,
481
+ 1,
482
+ -1,
483
+ 0,
484
+ 0,
485
+ -1,
486
+ -1,
487
+ 0,
488
+ -1,
489
+ -1,
490
+ -1,
491
+ -1,
492
+ -1,
493
+ -1,
494
+ -1,
495
+ -1,
496
+ 1,
497
+ -1,
498
+ 1,
499
+ -1,
500
+ -1,
501
+ 0,
502
+ -1,
503
+ -1,
504
+ -1,
505
+ 0,
506
+ -1,
507
+ -1,
508
+ -1,
509
+ -1,
510
+ -1,
511
+ 2,
512
+ -1,
513
+ -1,
514
+ -1,
515
+ -1,
516
+ 0,
517
+ -1,
518
+ 3,
519
+ 3,
520
+ 1,
521
+ 0,
522
+ -1,
523
+ -1,
524
+ 1,
525
+ -1,
526
+ -1,
527
+ -1,
528
+ -1,
529
+ -1,
530
+ 2,
531
+ -1,
532
+ 0,
533
+ 2,
534
+ -1,
535
+ -1,
536
+ -1,
537
+ -1,
538
+ 0,
539
+ 2,
540
+ 3,
541
+ -1,
542
+ -1,
543
+ 0,
544
+ -1,
545
+ -1,
546
+ -1,
547
+ -1,
548
+ 3,
549
+ -1,
550
+ 0,
551
+ 0,
552
+ -1,
553
+ -1,
554
+ 0,
555
+ -1,
556
+ -1,
557
+ -1,
558
+ 0,
559
+ -1,
560
+ -1,
561
+ -1,
562
+ 0,
563
+ -1,
564
+ 0,
565
+ 0,
566
+ -1,
567
+ -1,
568
+ 0,
569
+ -1,
570
+ -1,
571
+ -1,
572
+ -1,
573
+ 0
574
+ ],
575
+ "topic_sizes": {
576
+ "0": 109,
577
+ "-1": 264,
578
+ "2": 29,
579
+ "1": 34,
580
+ "3": 23
581
+ },
582
+ "topic_mapper": [
583
+ [
584
+ -1,
585
+ -1,
586
+ -1
587
+ ],
588
+ [
589
+ 0,
590
+ 0,
591
+ 0
592
+ ],
593
+ [
594
+ 1,
595
+ 1,
596
+ 3
597
+ ],
598
+ [
599
+ 2,
600
+ 2,
601
+ 2
602
+ ],
603
+ [
604
+ 3,
605
+ 3,
606
+ 1
607
+ ]
608
+ ],
609
+ "topic_labels": {
610
+ "-1": "-1_protests_tyranny_fauci_everyone",
611
+ "0": "0_vaccinated_fauci_deaths_2021",
612
+ "1": "1_guadeloupe_protests_toulon_genova",
613
+ "2": "2_austria_compulsory_vaccinated_restrictions",
614
+ "3": "3_australia_tyranny_protest_leader"
615
+ },
616
+ "custom_labels": null,
617
+ "_outliers": 1,
618
+ "topic_aspects": {}
619
+ }