有没有用GridSearchCV实现回调的标准方法？

Question

我喜欢使用GridSearchCV进行机器学习模型的超参数调优(这里主要使用sklearn )。

有没有办法传递一个函数/lambda作为回调，在每次搜索组合后都会被调用？我需要它来添加自定义日志记录，甚至在某些情况下发送事件。

事实上，我正在寻找一种类似于Keras回调的模式，其中每个回调都在每个时期之后执行。

感谢您的见解

Answer 1

我正在寻找在我的回调中获取当前参数的方法，并找到了您的问题，希望这能对某些人有所帮助。

grid = GridSearchCV(estimator=model, param_grid=param_grid, verbose=0, n_jobs=1)
grid_result = grid.fit(X_train, Y_train, callbacks=[YourCallback()])