python如何使用learning_curve

环球青藤 2020/11/22 23:41
python learning_curve函数

这个函数的作用为:对于不同大小的训练集,确定交叉验证训练和测试的分数。

一个交叉验证发生器将整个数据集分割k次,分割成训练集和测试集。(推荐学习:Python视频教程)

不同大小的训练集的子集将会被用来训练评估器并且对于每一个大小的训练子集都会产生一个分数,然后测试集的分数也会计算。然后,对于每一个训练子集,运行k次之后的所有这些分数将会被平均。

这个函数需要引用sklearn包

import sklearnfrom sklearn.learning_curve import learning_curve

这个函数的调用格式是:

learning_curve(estimator, X, y, train_sizes=array([ 0.1  ,  0.325,  0.55 ,  0.775,  1.   ]), cv=None, scoring=None, exploit_incremental_learning=False, n_jobs=1, pre_dispatch='all', verbose=0)

estimator:所使用的分类器

X:array-pke, shape (n_samples, n_features)

训练向量,n_samples是样本的数量,n_features是特征的数量

y:array-pke, shape (n_samples) or (n_samples, n_features), optional

目标相对于X分类或者回归

train_sizes:array-pke, shape (n_ticks,), dtype float or int

训练样本的相对的或绝对的数字,这些量的样本将会生成learning curve。如果dtype是float,他将会被视为最大数量训练集的一部分(这个由所选择的验证方法所决定)。否则,他将会被视为训练集的绝对尺寸。要注意的是,对于分类而言,样本的大小必须要充分大,达到对于每一个分类都至少包含一个样本的情况。

cv:int, cross-vapdation generator or an iterable, optional

确定交叉验证的分离策略

--None,使用默认的3-fold cross-vapdation,

--integer,确定是几折交叉验证

--一个作为交叉验证生成器的对象

--一个被应用于训练/测试分离的迭代器

verbose : integer, optional

控制冗余:越高,有越多的信息

返回值:

train_sizes_abs:array, shape = (n_unique_ticks,), dtype int

用于生成learning curve的训练集的样本数。由于重复的输入将会被删除,所以ticks可能会少于n_ticks.

train_scores : array, shape (n_ticks, n_cv_folds)

在训练集上的分数

test_scores : array, shape (n_ticks, n_cv_folds)

在测试集上的分数

更多Python相关技术文章,请访问Python教程栏目进行学习!

以上就是小编分享的关于python如何使用learning_curve的详细内容希望对大家有所帮助,更多有关python教程请关注环球青藤其它相关文章!

免费直播

    精选课程 更多

    注册电脑版

    版权所有 2003-2020 广州环球青藤科技发展有限公司