Scikit-Learn API的封装器
有两个封装器可用:
keras.wrappers.scikit_learn.KerasRegressor(build_fn=None, **sk_params)
, 这实现了Scikit-Learn regressor界面。
- build_fn: 可调用函数或类实例
- 函数
- 实现
__call__
函数的类的实例
sk_params
同时包含模型参数和拟合参数。 法律模型参数是build_fn
的参数。 类似于其他所有
估计者在Scikit-Learn, build_fn
应该为其参数提供默认值,这样就可以创建估计器,而不需要将任何值传递给sk_params
。
- 传递给
fit
,predict
,predict_proba
和score
函数的字典参数的值 - 传递给的值
当scikit-learn使用grid_search
API时,有效参数与sk_params
相同,包括拟合参数。
换句话说,你可以使用grid_search
来搜索最好的batch_size
或epochs
以及模型参数。