tf.keras.wrappers.scikit_learn.KerasRegressor

View source on GitHub

Class KerasRegressor

Implementation of the scikit-learn regressor API for Keras.

Aliases:

__init__

View source

__init__(
    build_fn=None,
    **sk_params
)

Initialize self. See help(type(self)) for accurate signature.

Methods

tf.keras.wrappers.scikit_learn.KerasRegressor.check_params

View source

check_params(params)

Checks for user typos in params.

Arguments:

  • params: dictionary; the parameters to be checked

Raises:

  • ValueError: if any member of params is not a valid argument.

tf.keras.wrappers.scikit_learn.KerasRegressor.filter_sk_params

View source

filter_sk_params(
    fn,
    override=None
)

Filters sk_params and returns those in fn's arguments.

Arguments:

  • fn: arbitrary function
  • override: dictionary, values to override sk_params

Returns:

  • res: dictionary containing variables in both sk_params and fn's arguments.

tf.keras.wrappers.scikit_learn.KerasRegressor.fit

View source

fit(
    x,
    y,
    **kwargs
)

Constructs a new model with build_fn & fit the model to (x, y).

Arguments:

  • x: array-like, shape (n_samples, n_features) Training samples where n_samples is the number of samples and n_features is the number of features.
  • y: array-like, shape (n_samples,) or (n_samples, n_outputs) True labels for x.
  • **kwargs: dictionary arguments Legal arguments are the arguments of Sequential.fit

Returns:

  • history: object details about the training history at each epoch.

tf.keras.wrappers.scikit_learn.KerasRegressor.get_params

View source

get_params(**params)

Gets parameters for this estimator.

Arguments:

  • **params: ignored (exists for API compatibility).

Returns:

Dictionary of parameter names mapped to their values.

tf.keras.wrappers.scikit_learn.KerasRegressor.predict

View source

predict(
    x,
    **kwargs
)

Returns predictions for the given test data.

Arguments:

  • x: array-like, shape (n_samples, n_features) Test samples where n_samples is the number of samples and n_features is the number of features.
  • **kwargs: dictionary arguments Legal arguments are the arguments of Sequential.predict.

Returns:

  • preds: array-like, shape (n_samples,) Predictions.

tf.keras.wrappers.scikit_learn.KerasRegressor.score

View source

score(
    x,
    y,
    **kwargs
)

Returns the mean loss on the given test data and labels.

Arguments:

  • x: array-like, shape (n_samples, n_features) Test samples where n_samples is the number of samples and n_features is the number of features.
  • y: array-like, shape (n_samples,) True labels for x.
  • **kwargs: dictionary arguments Legal arguments are the arguments of Sequential.evaluate.

Returns:

  • score: float Mean accuracy of predictions on x wrt. y.

tf.keras.wrappers.scikit_learn.KerasRegressor.set_params

View source

set_params(**params)

Sets the parameters of this estimator.

Arguments:

  • **params: Dictionary of parameter names mapped to their values.

Returns:

self