View source on GitHub |
Scikit learn wrapper for TensorFlow Learn Estimator.
tf.contrib.learn.SKCompat(
estimator
)
THIS CLASS IS DEPRECATED. See contrib/learn/README.md for general migration instructions.
Methods
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 |
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, name=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. |