8000 Update doctest expected result · scikit-learn/scikit-learn@e395131 · GitHub
[go: up one dir, main page]

Skip to content

Commit e395131

Browse files
authored
Update doctest expected result
1 parent 6e6250a commit e395131

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

sklearn/cluster/k_means_.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -871,12 +871,12 @@ class KMeans(BaseEstimator, ClusterMixin, TransformerMixin):
871871
... [10, 2], [10, 4], [10, 0]])
872872
>>> kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
873873
>>> kmeans.labels_
874-
array([0, 0, 0, 1, 1, 1], dtype=int32)
875-
>>> kmeans.predict([[0, 0], [4, 4]])
876-
array([0, 1], dtype=int32)
874+
array([1, 1, 1, 0, 0, 0], dtype=int32)
875+
>>> kmeans.predict([[0, 0], [12, 3]])
876+
array([1, 0], dtype=int32)
877877
>>> kmeans.cluster_centers_
878-
array([[1., 2.],
879-
[4., 2.]])
878+
array([[10., 2.],
879+
[1., 2.]])
880880
881881
See also
882882
--------

0 commit comments

Comments
 (0)
0