改善深层神经网络第二周-Optimitation and methods 错误修改

错误发生在此处：

5.1 - Mini-batch Gradient descent

layers_dims = [train_X.shape[0], 5, 2, 1]
parameters = model(train_X, train_Y, layers_dims, optimizer = "gd")


# Predict
predictions = predict(train_X, train_Y, parameters)


# Plot decision boundary
plt.title("Model with Gradient Descent optimization")
axes = plt.gca()
axes.set_xlim([-1.5,2.5])
axes.set_ylim([-1,1.5])
plot_decision_boundary(lambda x: predict_dec(parameters, x.T), train_X, train_Y)

错误提示：

ValueError: c of shape (1, 300) not acceptable as a color sequence for x with size 300, y with size 300

找到对应的函数：

意思是c的矩阵维度不对，在opt_utils.py脚本中找到plot_decision_boundary(model,X,y)函数，

将 plt.scatter(X[0, :], X[1, :], c=y, cmap=plt.cm.Spectral)改成：

plt.scatter(X[0, :], X[1, :], c=y.reshape(X[0,:].shape), cmap=plt.cm.Spectral)