lbhjvh14 commited on
Commit
bc6b184
·
1 Parent(s): b6cec09

Add BERTopic model

Browse files
Files changed (3) hide show
  1. config.json +3 -2
  2. topic_embeddings.safetensors +1 -1
  3. topics.json +0 -0
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "calculate_probabilities": false,
3
- "language": null,
4
  "low_memory": false,
5
  "min_topic_size": 10,
6
  "n_gram_range": [
@@ -10,5 +10,6 @@
10
  "nr_topics": 30,
11
  "seed_topic_list": null,
12
  "top_n_words": 10,
13
- "verbose": true
 
14
  }
 
1
  {
2
  "calculate_probabilities": false,
3
+ "language": "english",
4
  "low_memory": false,
5
  "min_topic_size": 10,
6
  "n_gram_range": [
 
10
  "nr_topics": 30,
11
  "seed_topic_list": null,
12
  "top_n_words": 10,
13
+ "verbose": true,
14
+ "embedding_model": "sentence-transformers/all-MiniLM-L6-v2"
15
  }
topic_embeddings.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6ee90bc952a4bb1c2e81c78199d02f00aee2972504a0f214e4ed8f485ccf8764
3
  size 46168
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:230c98a8949471affc7c8dbe869745f3ade8c18919a629ab468f818e80089444
3
  size 46168
topics.json CHANGED
The diff for this file is too large to render. See raw diff