在Scikit学习的任何模型收敛之前,是否有一种简便的方法来获取它们的预测?我的目的是制作模型动画,以便他们进行训练以观察其行为。
在理想情况下,这类似于在Keras训练ANN一段时间后如何提取结果。
答案 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
)