tanish78 commited on
Commit
d4dc4a9
·
verified ·
1 Parent(s): 65513e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -141,7 +141,7 @@ def silhouette_analysis(X, labels, num_clusters):
141
  fig.set_size_inches(10, 6)
142
 
143
  ax1.set_xlim([-0.1, 1])
144
- ax1.set_ylim([0, len(X.toarray()) + (num_clusters + 1) * 10])
145
 
146
  sample_silhouette_values = silhouette_samples(X, labels)
147
  y_lower = 10
@@ -193,8 +193,8 @@ def main(file, num_clusters_to_display):
193
  df['Cluster'] = pd.Categorical(df['Cluster'], categories=top_clusters, ordered=True)
194
  df = df.sort_values('Cluster')
195
 
196
- silhouette_avg = silhouette_score(X.toarray(), kmeans.labels_)
197
- silhouette_plot = silhouette_analysis(X.toarray(), kmeans.labels_, num_clusters=15)
198
 
199
  with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
200
  df.to_csv(tmpfile.name, index=False)
 
141
  fig.set_size_inches(10, 6)
142
 
143
  ax1.set_xlim([-0.1, 1])
144
+ ax1.set_ylim([0, X.shape[0] + (num_clusters + 1) * 10])
145
 
146
  sample_silhouette_values = silhouette_samples(X, labels)
147
  y_lower = 10
 
193
  df['Cluster'] = pd.Categorical(df['Cluster'], categories=top_clusters, ordered=True)
194
  df = df.sort_values('Cluster')
195
 
196
+ silhouette_avg = silhouette_score(X, kmeans.labels_)
197
+ silhouette_plot = silhouette_analysis(X, kmeans.labels_, num_clusters=15)
198
 
199
  with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
200
  df.to_csv(tmpfile.name, index=False)