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