tanish78 commited on
Commit
ace8a63
·
verified ·
1 Parent(s): f9c13b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -153,6 +153,15 @@ def cluster_data(df, num_clusters=5):
153
  return df
154
 
155
 
 
 
 
 
 
 
 
 
 
156
 
157
  def main(file, num_clusters):
158
  try:
 
153
  return df
154
 
155
 
156
+ def visualize_clusters(df):
157
+ plt.figure(figsize=(10, 6))
158
+ scatter = plt.scatter(df['PCA1'], df['PCA2'], c=df['Cluster'], cmap='viridis')
159
+ plt.legend(*scatter.legend_elements(), title="Clusters")
160
+ plt.title('Clusters of User Queries')
161
+ plt.xlabel('PCA Component 1')
162
+ plt.ylabel('PCA Component 2')
163
+ plt.show()
164
+
165
 
166
  def main(file, num_clusters):
167
  try: