experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Expose serialization/deserialization for feature_column as tf.__internal__ API.

Browse Source 
PiperOrigin-RevId: 361229619
Change-Id: Iffbce33ed89e97940819514ebcc6b305441d9b29
This commit is contained in:
Scott Zhu 2021-03-05 14:49:46 -08:00 committed by TensorFlower Gardener
parent 1a8561e57a
commit 88de667855
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/feature_column/serialization.py
View File

@ -25,6 +25,7 @@ from tensorflow.python.feature_column import sequence_feature_column as sfc_lib
from tensorflow.python.ops import init_ops from tensorflow.python.ops import init_ops
from tensorflow.python.util import tf_decorator from tensorflow.python.util import tf_decorator
from tensorflow.python.util import tf_inspect from tensorflow.python.util import tf_inspect
from tensorflow.python.util.tf_export import tf_export
_FEATURE_COLUMNS = [ _FEATURE_COLUMNS = [
@ -38,6 +39,7 @@ _FEATURE_COLUMNS = [
] ]
@tf_export('__internal__.feature_column.serialize_feature_column', v1=[])
def serialize_feature_column(fc): def serialize_feature_column(fc):
"""Serializes a FeatureColumn or a raw string key. """Serializes a FeatureColumn or a raw string key.
@ -86,6 +88,7 @@ def serialize_feature_column(fc):
raise ValueError('Instance: {} is not a FeatureColumn'.format(fc)) raise ValueError('Instance: {} is not a FeatureColumn'.format(fc))
@tf_export('__internal__.feature_column.deserialize_feature_column', v1=[])
def deserialize_feature_column(config, def deserialize_feature_column(config,
custom_objects=None, custom_objects=None,
columns_by_name=None): columns_by_name=None):
@ -94,7 +97,6 @@ def deserialize_feature_column(config,
This method should only be used to deserialize parent FeatureColumns when This method should only be used to deserialize parent FeatureColumns when
implementing FeatureColumn.from_config(), else deserialize_feature_columns() implementing FeatureColumn.from_config(), else deserialize_feature_columns()
is preferable. Returns a FeatureColumn for this config. is preferable. Returns a FeatureColumn for this config.
TODO(b/118939620): Simplify code if Keras utils support object deduping.
Args: Args:
config: A Dict with the serialization of feature columns acquired by config: A Dict with the serialization of feature columns acquired by
@ -111,6 +113,7 @@ def deserialize_feature_column(config,
Returns: Returns:
A FeatureColumn corresponding to the input `config`. A FeatureColumn corresponding to the input `config`.
""" """
# TODO(b/118939620): Simplify code if Keras utils support object deduping.
if isinstance(config, six.string_types): if isinstance(config, six.string_types):
return config return config
# A dict from class_name to class for all FeatureColumns in this module. # A dict from class_name to class for all FeatureColumns in this module.

8
tensorflow/tools/api/golden/v2/tensorflow.__internal__.feature_column.pbtxt
View File

@ -4,4 +4,12 @@ tf_module {
name: "StateManager" name: "StateManager"
mtype: "<type \'type\'>" mtype: "<type \'type\'>"
} }
member_method {
name: "deserialize_feature_column"
argspec: "args=[\'config\', \'custom_objects\', \'columns_by_name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
}
member_method {
name: "serialize_feature_column"
argspec: "args=[\'fc\'], varargs=None, keywords=None, defaults=None"
}
} }