模型进度 Scikit Learn

Question

是否有一种简单的方法可以在 Scikit 中的任何模型收敛之前获得预测？ 我的目标是制作模型的动画，因为他们训练观察他们的行为。

理想情况下，这类似于在 Keras 中训练 ANN 的一个时期之后如何提取结果。

Answer 1

您可能希望使用学习曲线类来评估中间结果。

在参数中，有一个名为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
                                               )