Wrappers for the Scikit-Learn API
There are two wrappers available:
keras.wrappers.scikit_learn.KerasRegressor(build_fn=None, **sk_params)
, which implements the Scikit-Learn regressor interface.
- build_fn: callable function or class instance
- A function
- An instance of a class that implements the
call
method
sk_params
could also accept parameters for calling fit
, predict
,predict_proba
, and score
methods (e.g., epochs
, batch_size
).fitting (predicting) parameters are selected in the following order:
- Values passed to the dictionary arguments of
fit
,predict
, , andscore
methods - Values passed to
sk_params