机器学习笔记(四)--sklearn数据集

sklearn数据集

 (一)机器学习的一般数据集会划分为两个部分

训练数据:用于训练,构建模型。

测试数据:在模型检验时使用,用于评估模型是否有效。

划分数据的API:sklearn.model_selection.train_test_split

示例代码如下:

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split


# 使用加载器读取数据并存入变量iris
iris = load_iris()
# 查验数据规模
print(iris.data.shape)
# 用于输入的数据
#print(iris.data)
# 标签
#print(iris.target)
# 对数据进行随机分割 测试数据占总数据的30% 训练数据占总数据的70%
x_train, x_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.3, random_state=30)
print('训练特征值:')
print(x_train)
print('训练目标值:')
print(y_train)
print('测试特征值:')
print(x_test)
print('测试目标值:')
print(y_test)

(二)转换器

fit_transform等价于  fit+transform

fit_transform()函数输入数据直接转换。

fit()函数输入数据,但不做事。计算平均值,方差等

transform()函数进行数据的转换。

(三)估计器

在sklearn中,估计器(estimator)是一类实现了算法的API。

1.用于分类的估计器:

sklearn.neighbors    k-近邻算法

sklearn.naive_bayes   贝叶斯

sklearn.linear_model.LogisticRegression  逻辑回归

sklearn.tree   决策树与随机森林

2.用于回归的估计器

sklearn.linear_model.LinearRegression   线性回归

sklearn.linear_model.Ridge   岭回归

猜你喜欢

转载自www.cnblogs.com/doctorXiong/p/10588809.html