from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
def datasets_demo():
'''sklearn数据集的使用'''
# 获取数据集
iris = load_iris()
print(type(iris))
print('鸢尾花数据集:\n', iris)
print('查看数据集描述:\n', iris['DESCR'])
print('查看数据集的特征名字:\n', iris.feature_names)
print('查看特征值:\n', iris.data)
print("鸢尾花的目标值:\n", iris.target)
# 对鸢尾花数据集进行分割
# 训练集的特征值x_train,测试集的特征值x_test,训练集的目标值y_train,测试集的目标值y_test
x_train, x_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.2, random_state=22)
print('x_train:\n', x_train, x_train.shape)
return None
if __name__ == '__main__':
datasets_demo()
sklearn数据集的使用
于 2022-04-01 21:09:09 首次发布