模型進度 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
                                               )