具体错误如下
ValueError: c of shape (1L, 300L) not acceptable as a color sequence for x with size 300, y with size 300
#代码
plot_decision_boundary(lambda x: predict_dec(parameters, x.T), train_X, train_Y)
#追踪
Traceback (most recent call last):
File "D:/Ng-DL-HW/course2/week1/initialization/zero_init.py", line 118, in <module>
plot_decision_boundary(lambda x: predict_dec(parameters, x.T), train_X, train_Y)
File "D:\Ng-DL-HW\course2\week1\initialization\init_utils.py", line 217, in plot_decision_boundary
plt.scatter(X[0, :], X[1, :], c=y, cmap=plt.cm.Spectral)
File "D:\tools\Anaconda2\lib\site-packages\matplotlib\pyplot.py", line 3357, in scatter
edgecolors=edgecolors, data=data, **kwargs)
File "D:\tools\Anaconda2\lib\site-packages\matplotlib\__init__.py", line 1710, in inner
return func(ax, *args, **kwargs)
File "D:\tools\Anaconda2\lib\site-packages\matplotlib\axes\_axes.py", line 4055, in scatter
raise ValueError(msg.format(c.shape, x.size, y.size))
ValueError: c of shape (1L, 300L) not acceptable as a color sequence for x with size 300, y with size 300
报错原因
此处train_Y应该reshape成一个一维向量,而不是作为一个多维矩阵送进去
解决
方法1:
plot_decision_boundary(lambda x: predict_dec(parameters, x.T), train_X, train_Y)
#改成
plot_decision_boundary(lambda x: predict_dec(parameters, x.T), train_X, np.squeeze(train_Y))
np.squeeze():
从数组的形状中删除单维条目,即把shape中为1的维度去掉
x = np.array([[[0], [1], [2]]]) x.shape Out[99]: (1, 3, 1) np.squeeze(x).shape Out[100]: (3,)