Skip to content

Instantly share code, notes, and snippets.

@hanneshapke

hanneshapke/model_input.py

Created March 20, 2020 18:56
Show Gist options
  • Save hanneshapke/7160e086a69861af28db67acb2b1408c to your computer and use it in GitHub Desktop.
Save hanneshapke/7160e086a69861af28db67acb2b1408c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_input.py
feature_spec = tf_transform_output.transformed_feature_spec()
feature_spec.pop(_LABEL_KEY)
inputs = {
key: tf.keras.layers.Input(
shape=(max_seq_length),
name=key,
dtype=tf.int32)
for key in feature_spec.keys()
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment