tf.contrib.learn.SKCompat
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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/learn/SKCompat