sklearn数据集使用（鸢尾花） 

 1     
 2 from sklearn.datasets import load_iris
 3 
 4     """
 5     sklearn数据集使用
 6     :return:
 7     """
 8     # 获取数据集
 9     iris = load_iris()
10     print("鸢尾花数据集：\n", iris)
11     print("查看数据集描述：\n", iris["DESCR"])
12     print("查看特征值的名字：\n", iris.feature_names)
13     print("查看特征值：\n", iris.data, iris.data.shape)
14 
15     # 训练集的特征值x_train 测试集的特征值x_test 训练集的目标值y_train 测试集的目标值y_test（x是特征，y是目标）
16     x_train, x_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.5, random_state=22)
17     print("训练集的特征值：\n", x_train, x_train.shape)
18 
19     return None

 

posted @ 2022-04-02 23:17  靠谱杨  阅读(294)  评论(0)  编辑  收藏  举报