使用scikit-learn的Iris数据集进行选取训练和测试集
时间: 2023-05-22 21:03:08 浏览: 166
python 模式识别 感知器.zip
可以使用以下代码来选取训练和测试集:
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
iris = load_iris()
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.2, random_state=0)
这个代码使用了scikit-learn库中的load_iris函数来加载Iris数据集,然后利用train_test_split函数将数据集分成了训练集和测试集,其中test_size参数用于指定测试集所占比例,random_state参数用于指定随机数种子,以确保每次运行代码时得到的训练集和测试集都是一致的。
阅读全文