![]() |
Utils for working with models.
Classes
class BatchReducibleBatchedDoFnWithModels
: Abstract class for DoFns that need the shared models.
class BatchReducibleDoFnWithModels
: Abstract class for DoFns that need the shared models.
class CombineFnWithModels
: Abstract class for CombineFns that need the shared models.
class DoFnWithModels
: Abstract class for DoFns that need the shared models.
class ModelContents
: Class for storing model contents.
class ModelSignaturesDoFn
: Updates extracts by calling specified model signature functions.
Functions
filter_by_input_names(...)
: Filters dict by input names.
find_input_name_in_features(...)
: Maps input name to an entry in features. Returns None if not found.
get_baseline_model_spec(...)
: Returns baseline model spec.
get_callable(...)
: Returns callable associated with given signature or None if not callable.
get_default_signature_name(...)
: Returns default signature name for given model.
get_feature_values_for_model_spec_field(...)
: Gets feature values associated with given model spec fields from extracts.
get_input_specs(...)
: Returns the input names and tensor specs associated with callable or None.
get_label_key(...)
: Returns the label_key corresponding to a given output name.
get_model_spec(...)
: Returns model spec with given model name.
get_model_type(...)
: Returns model type for given model spec taking into account defaults.
get_non_baseline_model_specs(...)
: Returns non-baseline model specs.
has_rubber_stamp(...)
: Check whether the candidate model is being rubber stamped.
input_specs_to_tensor_representations(...)
: Converts input specs into tensor representations.
is_callable_fn(...)
: Returns true if function is callable.
model_construct_fn(...)
: Returns function for constructing shared models.
verify_and_update_eval_shared_models(...)
: Verifies eval shared models and normnalizes to produce a single list.
Other Members | |
---|---|
KERAS_INPUT_SUFFIX |
'_input'
|