在 Jupyter Notebook 中查看 GridSearchCV 的进度

在 Jupyter Notebook 中是否可以查看 GridSearchCV 的进度?我正在运行以下 Python 脚本:

param_grid = {'learning_rate': [0.05, 0.10, 0.15, 0.20, 0.25, 0.30] ,'max_depth'        : [3, 4, 5, 6, 8, 10, 12, 15],'min_child_weight' : [1, 3, 5, 7],'gamma'            : [0.0, 0.1, 0.2 , 0.3, 0.4],'colsample_bytree' : [0.3, 0.4, 0.5 , 0.7],'verbose'          : [100] }xgboost_reg = XGBRegressor()grid_search = GridSearchCV(xgboost_reg, param_grid, cv=5, scoring='neg_mean_squared_error', return_train_score=True)grid_search.fit(my_data, my_labels, verbose=False)

我在单元格输出中只能看到一些警告信息。


回答:

您需要使用 verbose 参数:

grid_search = GridSearchCV(xgboost_reg, param_grid, cv=5, scoring='neg_mean_squared_error', return_train_score=True, verbose=2)grid_search.fit(my_data, my_labels, verbose=False)

我在测试数据上得到的示例输出如下:

Fitting 3 folds for each of 5 candidates, totalling 15 fits[CV] C=0.1 ...........................................................[CV] ............................................ C=0.1, total=   0.0s[CV] C=0.1 ...........................................................[CV] ............................................ C=0.1, total=   0.0s[CV] C=0.1 ...........................................................[CV] ............................................ C=0.1, total=   0.0s[CV] C=0.5 ...........................................................[CV] ............................................ C=0.5, total=   0.0s[CV] C=0.5 ...........................................................[CV] ............................................ C=0.5, total=   0.0s[CV] C=0.5 ...........................................................[CV] ............................................ C=0.5, total=   0.0s

Related Posts

多维度Top-k评分

例如,在机器学习中的自然语言处理中,通常使用波束搜索来…

R – Caret train() “错误:停止” 与 “在newdata中未找到对象中使用的所有变量名”

我正在尝试为蘑菇数据构建一个简单的朴素贝叶斯分类器。我…

创建训练和测试数据集分割,数据嵌套在多个文件夹中

我正在准备数据以训练一个图像识别模型。目前我有一个文件…

我的神经网络预测出现错误:IndexError: list index out of range

我正在进行一个简单的垃圾邮件/非垃圾邮件文本分类。我的…

python 给出数组是一维的,但索引了两个维度错误

我已经为 miniBatchGradientDesce…

TensorFlow自定义训练步骤使用不同的损失函数

背景 根据TensorFlow文档,可以使用以下方式执…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注