Scikit-Learn接口包装器
目前,有两个包装器可用:
keras.wrappers.scikit_learn.KerasClassifier(build_fn=None, **sk_params)
实现了sklearn的分类器接口
keras.wrappers.scikit_learn.KerasRegressor(build_fn=None, **sk_params)
实现了sklearn的回归器接口
build_fn:可调用的函数或类对象
build_fn
应构造、编译并返回一个Keras模型,该模型将稍后用于训练/测试。build_fn
的值可能为下列三种之一:
一个函数
一个具有
call
方法的类对象
sk_params
还接受用于调用fit
,predict
,predict_proba
和score
方法的参数,如nb_epoch
,batch_size
等。这些用于训练或预测的参数按如下顺序选择:
传递给
fit
,predict
,predict_proba
和的字典参数传递个
sk_params
的参数