Changing the clustering code

parent 367b5281
...@@ -191,7 +191,7 @@ class TopicCluster: ...@@ -191,7 +191,7 @@ class TopicCluster:
print("Cluster {}: {}".format(cat, ' '.join(top_ten_words))) print("Cluster {}: {}".format(cat, ' '.join(top_ten_words)))
print("\n") print("\n")
print(cluster_names_list) print(cluster_names_list)
# matching the clustering output
newkmlist_ = km.labels_ newkmlist_ = km.labels_
for tuple in cluster_names_list: for tuple in cluster_names_list:
newkmlist_ = [tuple[1] if i == tuple[0] else i for i in newkmlist_] newkmlist_ = [tuple[1] if i == tuple[0] else i for i in newkmlist_]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment