Interface for input creator.

The InputCreator class is an abstract class to implement create_inputs in ModelBuilder in tfr.keras.

To be implemented by subclasses:

  • __call__(): Contains the logic to create tf.keras.Input for context and example inputs.

Example subclass implementation:

class SimpleInputCreator(InputCreator):

  def __call__(self):
    return {}, {"example_feature_1": tf.keras.Input((1,), dtype=tf.float32)}



View source

Invokes the InputCreator instance.

A tuple of two dicts which map the context and example feature keys to the corresponding tf.keras.Input.