|
--- |
|
pipeline_tag: text-classification |
|
--- |
|
|
|
## Usage |
|
|
|
To use this model, please install BERTopic: |
|
|
|
Bash: pip install bertopic |
|
|
|
You can use the model as follows: |
|
|
|
from bertopic import BERTopic |
|
|
|
topic_model = BERTopic.load("Alprocco/Swiss_bertopic") |
|
|
|
topic_model.get_topic_info() |
|
|
|
## Topic overview |
|
|
|
Number of topics: 30 |
|
|
|
Training hyperparameters |
|
|
|
calculate_probabilities: False |
|
|
|
language: multilingual |
|
|
|
low_memory: False |
|
|
|
min_topic_size: 10 |
|
|
|
n_gram_range: (1, 1) |
|
|
|
nr_topics: 30 |
|
|
|
top_n_words: 10 |
|
|
|
verbose: True |
|
|
|
Note: When saving the model, make sure to also keep track of the versions of dependencies and Python used. |
|
Loading and saving the model should be done using the same dependencies and Python. |
|
|
|
Moreover, models saved in one version of BERTopic are not guaranteed to load in other versions. |
|
|
|
## Framework versions |
|
|
|
bertopic 0.15.0 |
|
|
|
Numpy: 1.24.4 |
|
|
|
HDBSCAN: 0.8.33 |
|
|
|
UMAP: 0.5.4 |
|
|
|
Pandas: 2.0.3 |
|
|
|
Scikit-Learn: 1.0.2 |
|
|
|
Sentence-transformers: 2.2.2 |
|
|
|
Transformers: 4.33.2 |
|
|
|
Numba: 0.58.0 |
|
|
|
Plotly: 5.17.0 |
|
|
|
Python: 3.8.10 |
