Aqsa Kausar
commited on
Update embedding_gen.py
Browse files- embedding_gen.py +1 -0
embedding_gen.py
CHANGED
|
@@ -90,6 +90,7 @@ def perform_kmeans_and_visualize(reduced_embeddings, skills, n_clusters, output_
|
|
| 90 |
base_folder = "./tags"
|
| 91 |
output_folder = "./plots"
|
| 92 |
specific_date = "03-01-2024" # Example date folder to process
|
|
|
|
| 93 |
|
| 94 |
# Load skills from the specified date folder
|
| 95 |
skills = load_skills_from_date(base_folder, specific_date)
|
|
|
|
| 90 |
base_folder = "./tags"
|
| 91 |
output_folder = "./plots"
|
| 92 |
specific_date = "03-01-2024" # Example date folder to process
|
| 93 |
+
n_clusters = 5
|
| 94 |
|
| 95 |
# Load skills from the specified date folder
|
| 96 |
skills = load_skills_from_date(base_folder, specific_date)
|