模型进度Scikit学习

时间:2019-11-23 01:52:14

标签: python scikit-learn

在Scikit学习的任何模型收敛之前,是否有一种简便的方法来获取它们的预测?我的目的是制作模型动画,以便他们进行训练以观察其行为。

在理想情况下,这类似于在Keras训练ANN一段时间后如何提取结果。

1 个答案:

答案 0 :(得分:1)

您可能想使用learning curve类来评估中间结果。

在参数中,有一个名为train_sizes的参数,可用于指定将用于生成曲线的训练示例的数量。它可以是绝对的,也可以是相对于可用样本的。

例如,使用线性回归:

from sklearn.linear_model import LinearRegression
from sklearn.model_selection import learning_curve
TRAIN_SIZES = [1, 100, 500, 2000, 5000]
train_sizes, train_scores, validation_scores = learning_curve(
                                                  estimator=LinearRegression(),
                                                  X=myX,
                                                  y=myY,
                                                  train_sizes=TRAIN_SIZES
                                               )