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还接受用于调用fitpredictpredict_probascore方法的参数,如nb_epochbatch_size等。这些用于训练或预测的参数按如下顺序选择:

    • 传递给fitpredictpredict_proba和的字典参数

    • 传递个sk_params的参数