machine-learning - 使用 tqdm 训练模型时检查进度

Question

我知道如何使用 tqdm 检查迭代进度：

for i in tqdm_notebook(range(100)):
time.sleep(0.1)

我想检查我的随机森林模型的训练进度。就像是：

//tqdm_notebook starts the progress bar 
RF_model=RandomForestRegressor(max_features='sqrt',n_estimators=100,oob_score=True)
RF_model.fit(x_train,y_train)
//tqdm_notebook stops the progress bar

Answer 1

您可以使用verbose相同的参数。

根据您的代码，只需添加一个参数：

RF_model=RandomForestRegressor(max_features='sqrt', n_estimators=100, oob_score=True, verbose=2) 
RF_model.fit(x_train,y_train)