D0men1c0's picture
Add BERTopic model
a0ddcbb verified
raw
history blame
352 Bytes
{
"calculate_probabilities": true,
"language": null,
"low_memory": false,
"min_topic_size": 10,
"n_gram_range": [
1,
4
],
"nr_topics": null,
"seed_topic_list": null,
"top_n_words": 10,
"verbose": true,
"zeroshot_min_similarity": 0.7,
"zeroshot_topic_list": null,
"embedding_model": "all-MiniLM-L6-v2"
}