scikit-learn 拟合剩余时间

Question

有没有办法在训练模型的时候估算剩余时间呢？比如说：

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

我有一个非常大的数据集（有几百万行），这会花费一些时间，所以我想知道大概还需要多久，这样我可以去做其他事情，等处理完成后再回来。

像随机森林这样的集成方法，估算剩余时间应该是比较简单的。

Answer 1

试试 verbose 选项。你可以把它设置为 0（没有输出）、1（每个任务更新一次）或者 2（每棵树更新一次），比如：

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