Skip to content

Commit 2279aab

Browse files
fix missing newaxis
1 parent 0d0546d commit 2279aab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scikitplot/decomposition.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def plot_pca_2d_projection(clf, X, y, title='PCA 2-D Projection',
177177
alpha=0.8, lw=2, label=label, color=color)
178178

179179
if label_dots:
180-
for dot in transformed_X[y == label, dimensions]:
180+
for dot in transformed_X[y == label, np.array(dimensions)[:, np.newaxis]]:
181181
ax.text(*dot, label)
182182

183183
if biplot:

0 commit comments

Comments
 (0)