初学python机器学习的forge数据集问题

初学python机器学习的forge数据集问题

刚开始接触机器学习,是按书上的代码一步步敲来着,但是遇到了一个问题始终过不去。
# 生成数据集
X, y = mglearn.datasets.make_forge()  # X和y是forge返回的两个特征
print("X.shape: {}".format(X.shape))  # X.shape: (26, 2)

# 数据集绘图
mglearn.discrete_scatter(X[:,0],X[:,1],y)  # 输入X第0列和第1列作为x轴,将y作为y轴
plt.legend(["Class 0", "Class 1"], loc=4)  # 设置图像的分类名称
plt.xlabel("First feature")   # 设置图像x轴的名称
plt.ylabel("Second feature")  # 设置图像的y轴的名称

这是书上的代码,跟网络上的资料也一致。但是运行时却出现
Traceback (most recent call last): File "G:/pythonProject/breast canser.py", line 3, in X, y = mglearn.datasets.make_forge() AttributeError: 'function' object has no attribute 'make_forge'

这项报错在查阅后,大家反映是对象的名称问题。但是下载以来没怎么动过,安装也是按站里的帖子一步步来的。
当然询问他人说是可能版本原因,但始终没有肯定的答复,希望大佬们能帮忙看看,感谢!

你可能感兴趣的:(机器学习,python)