From e9f3d777a03e6741e1f48b6090adc9fa96e63e86 Mon Sep 17 00:00:00 2001 From: Matt Watson Date: Thu, 21 Jan 2021 12:12:42 -0800 Subject: [PATCH] Set '_X_' as the default argument for category crossing separator PiperOrigin-RevId: 353072804 Change-Id: I4329b87ffe4b8d644768ba7e0caf5900eef4c7d4 --- .../python/keras/layers/preprocessing/category_crossing.py | 4 +--- ...layers.experimental.preprocessing.-category-crossing.pbtxt | 2 +- ...layers.experimental.preprocessing.-category-crossing.pbtxt | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/tensorflow/python/keras/layers/preprocessing/category_crossing.py b/tensorflow/python/keras/layers/preprocessing/category_crossing.py index 7cc6db3c16b..f5f9a918a58 100644 --- a/tensorflow/python/keras/layers/preprocessing/category_crossing.py +++ b/tensorflow/python/keras/layers/preprocessing/category_crossing.py @@ -114,13 +114,11 @@ class CategoryCrossing(base_preprocessing_layer.PreprocessingLayer): `[[b'1_X_2_X_3'], [b'4_X_5_X_6']]` """ - def __init__(self, depth=None, name=None, separator=None, **kwargs): + def __init__(self, depth=None, name=None, separator='_X_', **kwargs): super(CategoryCrossing, self).__init__(name=name, **kwargs) base_preprocessing_layer.keras_kpl_gauge.get_cell( 'CategoryCrossing').set(True) self.depth = depth - if separator is None: - separator = '_X_' self.separator = separator if isinstance(depth, (tuple, list)): self._depth_tuple = depth diff --git a/tensorflow/tools/api/golden/v1/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt index 575e3e35f08..e66d46d4260 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt @@ -138,7 +138,7 @@ tf_class { } member_method { name: "__init__" - argspec: "args=[\'self\', \'depth\', \'name\', \'separator\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\', \'None\'], " + argspec: "args=[\'self\', \'depth\', \'name\', \'separator\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\', \'_X_\'], " } member_method { name: "adapt" diff --git a/tensorflow/tools/api/golden/v2/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt index 575e3e35f08..e66d46d4260 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.keras.layers.experimental.preprocessing.-category-crossing.pbtxt @@ -138,7 +138,7 @@ tf_class { } member_method { name: "__init__" - argspec: "args=[\'self\', \'depth\', \'name\', \'separator\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\', \'None\'], " + argspec: "args=[\'self\', \'depth\', \'name\', \'separator\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\', \'_X_\'], " } member_method { name: "adapt"