backward compatible np.newaxis

bhsu · bhsu · commit 3e788426da2d · 2015-02-07T15:39:01.000-05:00
diff --git a/sklearn/cluster/k_means_.py b/sklearn/cluster/k_means_.py
@@ -1073,7 +1073,7 @@ def _mini_batch_step(X, x_squared_norms, centers, counts,
             centers[center_idx] *= counts[center_idx]
 
             # inplace sum with new points members of this cluster
-            weighted_X = np.multiply(sample_weight[center_mask, np.newaxis], X[center_mask])
+            weighted_X = np.multiply(sample_weight[center_mask][:, np.newaxis], X[center_mask])
             centers[center_idx] += np.sum(weighted_X, axis=0)
 
             # update the count statistics for this center
