机器学习001-数据的划分加载和介绍

数据集的划分

机器学习的数据一般 会划分为两个部分“
训练集:用于训练,构建模型
测试集:在检验模型时使用,用于评估模型是否有效
一般数据集划分75% 训练集划分25%
1.sklearn数据集划分api

sklearn.model_selection.train_test_split

2.sklearn.datasets加载获取流行的数据集
获取小规模的数据集数据包括在datasets里

datasets.load_*()

获取大规模的数据集,需要从网络上下载,第一个参数是data_home,表示数据的下载目录,默认是~/scikit_learn_data/

datasets.fetch_*(data_home=None)

两种api返回的的数据都是字典格式的

3.fetch_20newsgroups加载新闻(分类)数据集

##加载新闻数据集分类的数据集 subset参数传入要不的类型 train 训练集 test 测试集 all全部
news =fetch_20newsgroups(subset="all")
print(news.data)
print(news.target)

完整代码

#coding:utf-8
#加载曳尾花的数据
from sklearn.datasets import load_iris, fetch_20newsgroups
#加载拆分api
from sklearn.model_selection import train_test_split
#实例化数据集对象
li = load_iris()

print("获取特征值")
print(li["data"])
print("目标值")
print(li.target)
print(li.DESCR)
#拆分 第一个参数传入特征值 第一个目标值 第三个为训练集的大小
#返回值训练集特征 测试集特征 训练集目标,测试值目标
#注意返回值 训练集 train 训练集特征x_train 训练集目标 y_train 测试集:test x_test y_test
x_train, x_test,  y_train, y_test = train_test_split(li.data, li.target, test_size=0.25)
print("训练集特征和目标", x_train, y_train)
print("测试集特征和目标", x_test, y_test)

##加载新闻数据集分类的数据集 subset参数传入要不的类型 train 训练集 test 测试集 all全部
news =fetch_20newsgroups(subset="all")
print(news.data)
print(news.target)

猜你喜欢

转载自blog.csdn.net/qq_38441692/article/details/88025757