diff options
| author |  Petros Mol <pmol@google.com> | 2017-04-25 09:39:38 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-25 10:51:24 -0700 |
| commit | 311734a965b374ceb7a22d3b7331477c2402edec (patch) | |
| tree | 2393b92bd0b5ae5a521ac0b06d314f63b7e4ea9e /tensorflow/contrib/kernel_methods | |
| parent | 314fd704f504c2d40efa5559d3234337adeedf7e (diff) | |
Fix bug with optimizer not propagated from KernelLinearClassifier constructor.
Change: 154190471
Diffstat (limited to 'tensorflow/contrib/kernel_methods')
| -rw-r--r-- | tensorflow/contrib/kernel_methods/python/kernel_estimators.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/kernel_methods/python/kernel_estimators.py b/tensorflow/contrib/kernel_methods/python/kernel_estimators.py index 0ae88b22b4..de7530231d 100644 --- a/tensorflow/contrib/kernel_methods/python/kernel_estimators.py +++ b/tensorflow/contrib/kernel_methods/python/kernel_estimators.py @@ -176,7 +176,7 @@ class _KernelEstimator(estimator.Estimator): "head": head, _FEATURE_COLUMNS: feature_columns or [], _OPTIMIZER: optimizer, - _KERNEL_MAPPERS: kernel_mappers + _KERNEL_MAPPERS: kernel_mappers, } super(_KernelEstimator, self).__init__( model_fn=_kernel_model_fn, @@ -303,6 +303,7 @@ class KernelLinearClassifier(_KernelEstimator): weight_column_name=weight_column_name, head=head_lib.multi_class_head( n_classes=n_classes, weight_column_name=weight_column_name), + optimizer=optimizer, kernel_mappers=kernel_mappers, config=config) |