contrib.learn.SKCompat

tf.contrib.learn.SKCompat

class tf.contrib.learn.SKCompat

Defined in tensorflow/contrib/learn/python/learn/estimators/estimator.py.

Scikit learn wrapper for TensorFlow Learn Estimator.

Methods

__init__

__init__(estimator)

fit

fit(
    x,
    y,
    batch_size=128,
    steps=None,
    max_steps=None,
    monitors=None
)

get_params

get_params(deep=True)

Get parameters for this estimator.

Args:

• deep: boolean, optional

  If True, will return the parameters for this estimator and contained subobjects that are estimators.

Returns:

params : mapping of string to any Parameter names mapped to their values.

predict

predict(
    x,
    batch_size=128,
    outputs=None
)

score

score(
    x,
    y,
    batch_size=128,
    steps=None,
    metrics=None
)

set_params

set_params(**params)

Set the parameters of this estimator.

The method works on simple estimators as well as on nested objects (such as pipelines). The former have parameters of the form <component>__<parameter> so that it's possible to update each component of a nested object.

Args:

**params: Parameters.

Returns:

self

Raises:

• ValueError: If params contain invalid names.