GitHub

tfma.model_util.get_input_specs

Returns the input names and tensor specs associated with callable or None.

tfma.model_util.get_input_specs(
    model: Any,
    signature_name: Optional[Text] = None,
    required: bool = True
) -> Optional[Dict[Text, tf.TypeSpec]]

Args
`model`	A model that is callable or contains a `signatures` attribute. If neither of these conditions are met, then None will be returned.
`signature_name`	Optional name of signature to use. If not provided then either the default serving signature will be used (if model is not callable) or the model itself will be used (if the model is callable). If provided then model.signatures will be used regardless of whether the model is callable or not.
`required`	True if signature_name is required to exist if provided.

Returns
Dict mapping input names to their associated tensor specs or None if no callable could be found.

Raises
`ValueError`	If signature_name not found in model.signatures.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2021-02-26 UTC.

tfma.model_util.get_input_specs

Args

Returns

Raises