mirror of
https://github.com/marvinscham/masterthesis-playground.git
synced 2026-03-22 08:22:43 +01:00
22.02.
This commit is contained in:
@@ -16,10 +16,10 @@ from bertopic import BERTopic
|
||||
|
||||
param_grid = {
|
||||
"n_gram_max": [2, 3], # Vectorization
|
||||
"min_document_frequency": [1], # Vectorization
|
||||
"min_document_frequency": [1, 2], # Vectorization
|
||||
"min_samples": [10, 25], # HDBSCAN
|
||||
"min_topic_size": [10, 20, 30, 40, 50], # HDBSCAN
|
||||
"n_neighbors": [15], # UMAP
|
||||
"min_topic_size": [100, 200], # HDBSCAN
|
||||
"n_neighbors": [15, 25], # UMAP
|
||||
"n_components": [2, 5], # UMAP
|
||||
"min_dist": [0.01, 0.1], # UMAP
|
||||
"nr_topics": ["auto"], # Topic Modeling
|
||||
|
||||
Reference in New Issue
Block a user