Add BERTopic model

README.md

@@ -0,0 +1,74 @@
+
+---
+tags:
+- bertopic
+library_name: bertopic
+pipeline_tag: text-classification
+---
+
+# rag-topic-model
+
+This is a [BERTopic](https://github.com/MaartenGr/BERTopic) model.
+BERTopic is a flexible and modular topic modeling framework that allows for the generation of easily interpretable topics from large datasets.
+
+## Usage
+
+To use this model, please install BERTopic:
+
+```
+pip install -U bertopic
+```
+
+You can use the model as follows:
+
+```python
+from bertopic import BERTopic
+topic_model = BERTopic.load("apostolosfilippas/rag-topic-model")
+
+topic_model.get_topic_info()
+```
+
+## Topic overview
+
+* Number of topics: 5
+* Number of training documents: 168
+
+<details>
+  <summary>Click here for an overview of all topics.</summary>
+
+  | Topic ID | Topic Keywords | Topic Frequency | Label | 
+|----------|----------------|-----------------|-------| 
+| -1 | for - klarna - to - my - the | 15 | -1_for_klarna_to_my | 
+| 0 | klarna - my - declined - for - ve | 51 | 0_klarna_my_declined_for | 
+| 1 | payment - to - the - my - for | 50 | 1_payment_to_the_my | 
+| 2 | my - klarna - and - details - account | 28 | 2_my_klarna_and_details | 
+| 3 | store - the - it - refund - back | 24 | 3_store_the_it_refund |
+
+</details>
+
+## Training hyperparameters
+
+* calculate_probabilities: False
+* language: None
+* low_memory: False
+* min_topic_size: 10
+* n_gram_range: (1, 1)
+* nr_topics: auto
+* seed_topic_list: None
+* top_n_words: 10
+* verbose: False
+* zeroshot_min_similarity: 0.7
+* zeroshot_topic_list: None
+
+## Framework versions
+
+* Numpy: 1.26.4
+* HDBSCAN: 0.8.40
+* UMAP: 0.5.7
+* Pandas: 2.2.3
+* Scikit-Learn: 1.6.1
+* Sentence-transformers: 3.4.1
+* Transformers: 4.50.1
+* Numba: 0.61.2
+* Plotly: 6.0.1
+* Python: 3.11.7

config.json

@@ -0,0 +1,16 @@
+{
+  "calculate_probabilities": false,
+  "language": null,
+  "low_memory": false,
+  "min_topic_size": 10,
+  "n_gram_range": [
+    1,
+    1
+  ],
+  "nr_topics": "auto",
+  "seed_topic_list": null,
+  "top_n_words": 10,
+  "verbose": false,
+  "zeroshot_min_similarity": 0.7,
+  "zeroshot_topic_list": null
+}

topic_embeddings.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:03b618de1995888187647d51b92bf5264397d9940827813be2bcad17dd3bc68c
+size 7768

topics.json

@@ -0,0 +1,433 @@
+{
+  "topic_representations": {
+    "-1": [
+      [
+        "for",
+        0.06602511651625986
+      ],
+      [
+        "klarna",
+        0.06438953438618288
+      ],
+      [
+        "to",
+        0.0588484734166664
+      ],
+      [
+        "my",
+        0.0574797691276483
+      ],
+      [
+        "the",
+        0.05650736581105164
+      ],
+      [
+        "but",
+        0.050391580561394415
+      ],
+      [
+        "payment",
+        0.04423445960277377
+      ],
+      [
+        "card",
+        0.04413108718210577
+      ],
+      [
+        "using",
+        0.04281002577647979
+      ],
+      [
+        "just",
+        0.039667471149400856
+      ]
+    ],
+    "0": [
+      [
+        "klarna",
+        0.0684502885090884
+      ],
+      [
+        "my",
+        0.06587602342538006
+      ],
+      [
+        "declined",
+        0.06268185270099286
+      ],
+      [
+        "for",
+        0.05184770062228036
+      ],
+      [
+        "ve",
+        0.0514886647186727
+      ],
+      [
+        "the",
+        0.050923634011540604
+      ],
+      [
+        "in",
+        0.05051278288225979
+      ],
+      [
+        "with",
+        0.04270783667311336
+      ],
+      [
+        "what",
+        0.040295476736352545
+      ],
+      [
+        "to",
+        0.039236097768212165
+      ]
+    ],
+    "1": [
+      [
+        "payment",
+        0.09882734039896089
+      ],
+      [
+        "to",
+        0.09353077326027964
+      ],
+      [
+        "the",
+        0.08362522226811411
+      ],
+      [
+        "my",
+        0.07677260183954467
+      ],
+      [
+        "for",
+        0.0694800029933506
+      ],
+      [
+        "due",
+        0.06434756124704745
+      ],
+      [
+        "is",
+        0.060984550473916356
+      ],
+      [
+        "pay",
+        0.05410201464634094
+      ],
+      [
+        "bought",
+        0.05257795446732636
+      ],
+      [
+        "card",
+        0.048551252035772596
+      ]
+    ],
+    "2": [
+      [
+        "my",
+        0.10820482052156193
+      ],
+      [
+        "klarna",
+        0.0778837462113562
+      ],
+      [
+        "and",
+        0.07387174047283641
+      ],
+      [
+        "details",
+        0.07121357997239698
+      ],
+      [
+        "account",
+        0.06107877265996873
+      ],
+      [
+        "call",
+        0.05786738490569826
+      ],
+      [
+        "card",
+        0.05459288080457491
+      ],
+      [
+        "someone",
+        0.05113457291213346
+      ],
+      [
+        "this",
+        0.051056373196725964
+      ],
+      [
+        "worried",
+        0.04984950598067789
+      ]
+    ],
+    "3": [
+      [
+        "store",
+        0.08068849143662853
+      ],
+      [
+        "the",
+        0.07418625377113114
+      ],
+      [
+        "it",
+        0.07287641056570636
+      ],
+      [
+        "refund",
+        0.06783625470050814
+      ],
+      [
+        "back",
+        0.06783625470050814
+      ],
+      [
+        "for",
+        0.06532535937718241
+      ],
+      [
+        "to",
+        0.06124252441610852
+      ],
+      [
+        "credit",
+        0.0593815717854626
+      ],
+      [
+        "ago",
+        0.05849200859752171
+      ],
+      [
+        "days",
+        0.05706008655023621
+      ]
+    ]
+  },
+  "topics": [
+    2,
+    3,
+    0,
+    2,
+    -1,
+    3,
+    -1,
+    -1,
+    -1,
+    -1,
+    -1,
+    0,
+    1,
+    -1,
+    1,
+    -1,
+    -1,
+    -1,
+    0,
+    -1,
+    3,
+    -1,
+    -1,
+    -1,
+    1,
+    -1,
+    -1,
+    -1,
+    0,
+    1,
+    1,
+    -1,
+    -1,
+    -1,
+    0,
+    3,
+    3,
+    3,
+    3,
+    0,
+    -1,
+    0,
+    -1,
+    1,
+    3,
+    1,
+    1,
+    -1,
+    -1,
+    0,
+    -1,
+    1,
+    -1,
+    -1,
+    -1,
+    -1,
+    2,
+    -1,
+    2,
+    1,
+    0,
+    2,
+    0,
+    -1,
+    0,
+    2,
+    -1,
+    -1,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2,
+    0,
+    2,
+    2,
+    2,
+    0,
+    2,
+    -1,
+    2,
+    2,
+    0,
+    2,
+    2,
+    -1,
+    -1,
+    -1,
+    -1,
+    0,
+    -1,
+    0,
+    3,
+    3,
+    3,
+    0,
+    3,
+    -1,
+    -1,
+    3,
+    3,
+    0,
+    0,
+    0,
+    1,
+    1,
+    1,
+    1,
+    -1,
+    1,
+    0,
+    0,
+    0,
+    0,
+    -1,
+    0,
+    0,
+    0,
+    0,
+    -1,
+    0,
+    -1,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    -1,
+    0,
+    0,
+    0,
+    0,
+    -1,
+    0,
+    0,
+    0,
+    -1,
+    0,
+    0,
+    0,
+    0,
+    0,
+    2,
+    1,
+    1,
+    0,
+    2,
+    -1,
+    1,
+    -1,
+    -1,
+    1,
+    1,
+    1,
+    1,
+    1,
+    1,
+    1,
+    1,
+    1,
+    3,
+    1
+  ],
+  "topic_sizes": {
+    "2": 24,
+    "3": 15,
+    "0": 50,
+    "-1": 51,
+    "1": 28
+  },
+  "topic_mapper": [
+    [
+      -1,
+      -1,
+      -1,
+      -1
+    ],
+    [
+      0,
+      0,
+      0,
+      2
+    ],
+    [
+      1,
+      1,
+      1,
+      0
+    ],
+    [
+      2,
+      2,
+      2,
+      3
+    ],
+    [
+      3,
+      3,
+      3,
+      1
+    ]
+  ],
+  "topic_labels": {
+    "-1": "-1_for_klarna_to_my",
+    "0": "0_klarna_my_declined_for",
+    "1": "1_payment_to_the_my",
+    "2": "2_my_klarna_and_details",
+    "3": "3_store_the_it_refund"
+  },
+  "custom_labels": null,
+  "_outliers": 1,
+  "topic_aspects": {}
+}