scikit-learn适合剩余时间

Question

有没有办法估算拟合模型时的剩余时间？例如

model = sk.ensemble.RandomForestRegressor(n_estimators=10)
model.fit(x, y)

我有一个非常大的数据集(数百万行),这需要一些时间,所以我想知道估计的时间,所以我可以做其他的thigngs并在过程结束时回来.

对随机森林等集合估计剩余时间应该[合理]容易.

Answer 1

尝试verbose选项.您可以将其从0(无输出),1(每个作业更新)和2(每个树的更新)更改,例如

model = RandomForestRegressor(n_estimators=100, verbose=2, n_jobs=2).fit(X_train, y_train)