runner.RunResult

View source on GitHub

Holds the return values of run(...).

runner.RunResult(
    preprocess_model: Optional[tf.keras.Model],
    base_model: tf.keras.Model,
    trained_model: tf.keras.Model
)

Attributes
preprocess_model	Keras model containing only the computation for preprocessing inputs. It is not trained. The model takes serialized GraphTensors as its inputs and returns preprocessed GraphTensors. None when no preprocess model exists.
base_model	Keras base GNN (as returned by the user provided model_fn). The model both takes and returns GraphTensors. The model contains any--but not all--trained weights. The trained_model contains all base_model trained weights in addition to any prediction trained weights.
trained_model	Keras model for the e2e GNN. (Base GNN plus any prediction head(s).) The model takes preprocess_model output as its inputs and returns Task predictions as its output. Output matches the structure of the Task: an atom for single- or a mapping for multi- Task training. The model contains all trained weights.

Methods

__eq__

__eq__(
    other
)

Return self==value.