Add BERTopic model

topic_embeddings.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e439d8ba00d95b1639268781b85578d0e0f259d096759a0c1fefef55b1f34cd6
+oid sha256:937437c975b8c124981aa4601071d8c60d00dd6ffd19c21ba764a966fc06fe44
 size 4696

topics.json

@@ -3,127 +3,127 @@
     "-1": [
       [
         "babi",
-        0.07951911004253617
-      ],
-      [
-        "not",
-        0.06306454796499554
+        0.09407191906887093
       ],
       [
         "hospit",
-        0.06138068683159471
+        0.07400895739061451
       ],
       [
         "day",
-        0.056176397342317844
+        0.061134561555409535
       ],
       [
-        "child",
-        0.05581198255233367
+        "mother",
+        0.049380834626509065
       ],
       [
-        "start",
-        0.04949500584243205
+        "not",
+        0.0479281217522935
       ],
       [
-        "mother",
-        0.04935982508148416
+        "birth",
+        0.046460659903828645
       ],
       [
-        "pregnanc",
-        0.043659084332731427
+        "went",
+        0.04557815437130582
       ],
       [
-        "birth",
-        0.04214803714021838
+        "said",
+        0.04557815437130582
       ],
       [
-        "pain",
-        0.040988450825326636
+        "month",
+        0.04534094691452953
+      ],
+      [
+        "after",
+        0.04498875913470191
       ]
     ],
     "0": [
       [
         "babi",
-        0.13755439950323678
+        0.11698491140244063
+      ],
+      [
+        "not",
+        0.07597649245620709
       ],
       [
         "mother",
-        0.07212192768340195
+        0.06582582392303368
       ],
       [
         "hospit",
-        0.07128042283007006
+        0.06230886048811066
       ],
       [
-        "not",
-        0.06910990000789143
+        "child",
+        0.05216887131949626
       ],
       [
         "told",
-        0.052424681246105796
+        0.05044451772083532
       ],
       [
-        "nurs",
-        0.04542419502143051
+        "doctor",
+        0.046417932146163146
       ],
       [
         "day",
-        0.043972466401590506
+        0.04127335623542829
       ],
       [
-        "after",
-        0.04348270127109151
-      ],
-      [
-        "went",
-        0.043073424266603946
+        "nurs",
+        0.04024313976797694
       ],
       [
-        "child",
-        0.04154407163333329
+        "clinic",
+        0.03817278670404981
       ]
     ],
     "1": [
       [
-        "babi",
-        0.13390355361647183
+        "breath",
+        0.13024178601741546
       ],
       [
-        "breath",
-        0.12700245221595585
+        "day",
+        0.11932204698650219
+      ],
+      [
+        "babi",
+        0.11725523403401376
       ],
       [
         "born",
-        0.11400605983676022
+        0.10315932944743283
       ],
       [
-        "day",
-        0.10549457040059432
+        "prematur",
+        0.08520406002589198
       ],
       [
         "problem",
-        0.0762199002789424
+        0.07758923553192947
       ],
       [
-        "prematur",
-        0.07323779258501976
+        "not",
+        0.06522923496608235
       ],
       [
-        "month",
-        0.06758757163505731
+        "child",
+        0.06262913632982825
       ],
       [
         "lung",
-        0.06192784709570467
+        0.05603592642060726
       ],
       [
         "die",
-        0.06063041997295525
-      ],
-      [
-        "not",
-        0.06007315822752551
+        0.0548619387204676
       ]
     ]
   },
@@ -131,23 +131,23 @@
     1,
     -1,
     1,
-    -1,
+    0,
     -1,
     -1,
     0,
     1,
-    -1,
-    1,
-    -1,
     0,
-    -1,
-    -1,
     1,
+    -1,
     0,
     0,
     0,
+    1,
+    -1,
+    -1,
+    -1,
+    -1,
     -1,
-    0,
     -1,
     -1,
     -1,
@@ -159,17 +159,17 @@
     0,
     -1,
     1,
-    0,
     -1,
+    0,
     -1,
     -1,
     0,
     0,
-    -1,
-    -1,
+    0,
+    0,
     0,
     -1,
-    -1,
+    0,
     0,
     0,
     0,
@@ -184,7 +184,7 @@
     0,
     0,
     0,
-    0,
+    -1,
     0,
     0,
     1,
@@ -195,12 +195,12 @@
     1,
     0,
     1,
-    -1,
-    -1,
+    0,
+    1,
     0,
     -1,
-    -1,
-    -1,
+    0,
+    0,
     -1,
     -1,
     0,
@@ -209,7 +209,7 @@
     -1,
     -1,
     -1,
-    1,
+    -1,
     0,
     0,
     -1,
@@ -220,7 +220,7 @@
     0,
     0,
     -1,
-    0,
+    -1,
     0,
     0,
     -1,
@@ -228,14 +228,14 @@
     1,
     0,
     0,
-    -1,
-    -1,
+    1,
+    0,
     0
   ],
   "topic_sizes": {
-    "1": 13,
-    "-1": 44,
-    "0": 46
+    "1": 14,
+    "-1": 38,
+    "0": 51
   },
   "topic_mapper": [
     [
@@ -258,9 +258,9 @@
     ]
   ],
   "topic_labels": {
-    "-1": "-1_babi_not_hospit_day",
-    "0": "0_babi_mother_hospit_not",
-    "1": "1_babi_breath_born_day"
+    "-1": "-1_babi_hospit_day_mother",
+    "0": "0_babi_not_mother_hospit",
+    "1": "1_breath_day_babi_born"
   },
   "custom_labels": null,
   "_outliers": 1,