欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

sklearn

程序员文章站 2022-06-12 22:51:35
...

1、sklearn.datasets.load_*()
获取小规模数据集,数据集包含在datasets里
ex_1:
sklearn.datasets.load_iris()返回鸢尾花的数据集
2、sklearn.datasets.fetch_*(data_home=None,subset='train/test/all')
获取大规模数据集,需要从网络上下载,传入的函数的第一个参数是data_home,表示数据集下载的目录,默认是~/scikit_learn_data/,第二个参数表示想要的数据集是训练集还是测试集,还是都要
3、数据集的返回值
load和fetch返回的是datasets.base.Bunch(继承自字典)
五个键:
data:数据集中的特值
target:数据集中的目标值
DESCR:数据集的描述
feature_names:特征的名字
target_names:目标值代表什么

	dict["key"] = values
    bunch.key = values

ex_2:

from sklearn.datasets import load_iris

def datasets_demo():

	'''
	sklearn数据集的使用
	'''
	iris = load_iris()
	print("鸢尾花数据集:\n", iris)

4、数据集划分
训练数据:用于训练,构建模型
划分比例:数据集20%~30%
数据集划分api

from sklearn.model_selection import train_test_split(arrays, *options)

x 数据集的特征值
y 数据集的标签值
test_size 测试集大小,一般为float,默认0.25
random_state 随机数种子
return 训练集特征值,测试集特征值,训练集目标值,测试集目标值

ex_3:

def datasets_demo():
    iris = load_iris()
    # print("鸢尾花数据集:\n", iris)
    # print("鸢尾花数据集描述:\n", iris["DESCR"])
    print("源数据:", iris.data.shape)
    # 数据集划分
    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,x_train.shape)
    return None
相关标签: sklearn