View source on GitHub |
InputCreator with feature specs.
Inherits From: InputCreator
tfr.keras.model.FeatureSpecInputCreator(
context_feature_spec: Dict[str, Union[tf.io.FixedLenFeature, tf.io.VarLenFeature, tf.io.
RaggedFeature]],
example_feature_spec: Dict[str, Union[tf.io.FixedLenFeature, tf.io.VarLenFeature, tf.io.
RaggedFeature]]
)
Example usage:
input_creator=FeatureSpecInputCreator(
{},
{"example_feature_1": tf.io.FixedLenFeature(
shape=(1,), dtype=tf.float32, default_value=0.0)})
Methods
__call__
__call__() -> Tuple[tfr.keras.model.TensorDict
, tfr.keras.model.TensorDict
]
See InputCreator
.