使用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)