tfma.model_util.model_construct_fn

View source on GitHub

Returns function for constructing shared models.

View aliases

Main aliases

tfma.model_construct_fn

tfma.model_util.model_construct_fn(
    eval_saved_model_path: Optional[Text] = None,
    add_metrics_callbacks: Optional[List[tfma.types.MetricVariablesType]] = None,
    include_default_metrics: Optional[bool] = None,
    additional_fetches: Optional[List[Text]] = None,
    blacklist_feature_fetches: Optional[List[Text]] = None,
    tags: Optional[List[Text]] = None,
    model_type: Optional[Text] = constants.TF_ESTIMATOR
) -> Callable[[], Any]