python - 散点图错误:数组索引过多
问题描述
我正在使用逻辑回归模型来预测购买产品的人的年龄。为了在图中显示,我正在使用 matplotlib 散点图,但是在采用参数时存在问题,这会导致索引过多的错误`
# Visualising the Training set results
from matplotlib.colors import ListedColormap
x_set, y_set = x_train, y_train
X1, X2 = np.meshgrid(np.arange(start = x_set[:, 0].min() - 1, stop = x_set[:, 0].max() + 1, step = 0.01),
np.arange(start = x_set[:, 1].min() - 1, stop = x_set[:, 1].max() + 1, step = 0.01))
plt.contourf(X1, X2, classifier.predict(np.array([X1.ravel(), X2.ravel()]).T).reshape(X1.shape),
alpha = 0.75, cmap = ListedColormap(('red', 'green')))
plt.xlim(X1.min(), X1.max())
plt.ylim(X2.min(), X2.max())
for i, j in enumerate(np.unique(y_set)):
plt.scatter(x_set[(y_set == j),0], x_set[(y_set == j),1],
c = ListedColormap(('red', 'green'))(i), label = j)
plt.title('Logistic Regression (Training set)')
plt.xlabel('Age')
plt.ylabel('Estimated Salary')
plt.legend()
plt.show()
解决方案
推荐阅读
- reactjs - Typescript args 类型推断的困难
- python - 在 python 文档中理解适当的 **kwargs
- c# - 使用对象值类型的子字典对字典进行排序
- javascript - 在输入中显示当前时间和灰色之前的时间
- android - MapView 在 Android mvvm 数据绑定中不显示谷歌地图 [已解决]
- javascript - Cheerio 无法获取文本数据
- oracle - 我有问题使用合并编译触发器
- docker - nginx 反向代理到一组 dockerized 应用程序
- spring-boot - 数据库更新问题 - 发布记录列表时,很少有记录没有更新 - 使用 Spring JPA、Jersey Rest、Postgresql DB
- sql-server - QSqlQuery 选择已删除的表