tylerandrewscott commited on
Commit
312e390
1 Parent(s): f8bab7b

Add BERTopic model

Browse files
Files changed (3) hide show
  1. config.json +2 -2
  2. topic_embeddings.safetensors +2 -2
  3. topics.json +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "calculate_probabilities": false,
3
- "language": "english",
4
  "low_memory": false,
5
  "min_topic_size": 10,
6
  "n_gram_range": [
@@ -13,5 +13,5 @@
13
  "verbose": false,
14
  "zeroshot_min_similarity": 0.7,
15
  "zeroshot_topic_list": null,
16
- "embedding_model": "sentence-transformers/all-MiniLM-L6-v2"
17
  }
 
1
  {
2
  "calculate_probabilities": false,
3
+ "language": null,
4
  "low_memory": false,
5
  "min_topic_size": 10,
6
  "n_gram_range": [
 
13
  "verbose": false,
14
  "zeroshot_min_similarity": 0.7,
15
  "zeroshot_topic_list": null,
16
+ "embedding_model": "thenlper/gte-small"
17
  }
topic_embeddings.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f2ddad342086507badd0affff06af72730cc2c26a15ea4cf32e02baff8b23373
3
- size 12043872
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a80ab49698f97a4a459a38b3cd5b03c1f6bcbf53a1633974c2766bd4e3cbabf
3
+ size 9306720
topics.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6350e1fb7090be1c5da4bf8b2dff088bac8ecfc71085de17eb08a08124e0c5b4
3
- size 11910304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ec555d6b5188ac9421012c58950d0542ff99df402e66e365e8398b70f2abbc1
3
+ size 10419121