Skip to content

Commit 94154b9

Browse files
committed
Enable more than one parallel job.
1 parent a2daa18 commit 94154b9

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Diff for: p314_k_means.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@
9191
title='k-means cluster E13B sum of columns')
9292

9393
from sklearn.cluster import KMeans
94-
km = KMeans(n_clusters=3,init='random',n_init=10,max_iter=300,tol=1e-04,random_state=0)
94+
km = KMeans(n_clusters=3,init='random',n_init=10,max_iter=300,tol=1e-04,random_state=0,n_jobs=-1)
9595
y_km = km.fit_predict(X)
9696

9797
legend=[]
@@ -111,7 +111,7 @@
111111
axis_labels = ['column {} sum'.format(columnsXY[i]) for i in range(len(columnsXY))],
112112
title='column sums k means centroids')
113113

114-
km = KMeans(n_clusters=3,n_init=10,max_iter=300,tol=1e-04,random_state=0)
114+
km = KMeans(n_clusters=3,n_init=10,max_iter=300,tol=1e-04,random_state=0,n_jobs=-1)
115115
y_km = km.fit_predict(X)
116116

117117
legend=[]
@@ -163,7 +163,7 @@
163163

164164
distortions=[]
165165
for i in range(1,30):
166-
km = KMeans(n_clusters=i,n_init=10,max_iter=300,tol=1e-04,random_state=0)
166+
km = KMeans(n_clusters=i,n_init=10,max_iter=300,tol=1e-04,random_state=0,n_jobs=-1)
167167
y_km = km.fit_predict(X_image)
168168
distortions.append(km.inertia_)
169169

@@ -175,7 +175,7 @@
175175
ocr_utils.show_figures(plt, title)
176176

177177

178-
km = KMeans(n_clusters=8,n_init=10,max_iter=300,tol=1e-04,random_state=0)
178+
km = KMeans(n_clusters=8,n_init=10,max_iter=300,tol=1e-04,random_state=0,n_jobs=-1)
179179
y_km = km.fit_predict(X_image)
180180

181181
nClusters = km.cluster_centers_.shape[0]

0 commit comments

Comments
 (0)