the-algorithm/twml/twml/contrib/parsers.py

22 lines
882 B
Python

'''
Contains implementations of functions to parse the contrib.FeatureConfig
Modelers can use the functions in this module as the the train/eval_parse_fn of
the DataRecordTrainer constructor to customize how to parse their datasets.
Modelers may also provide custom implementations of train/eval_parse_fn using these as reference.
'''
from twitter.deepbird.io.legacy.contrib.parsers import (
_convert_to_fixed_length_tensor, # noqa: F401
_get_input_receiver_fn_feature_dict, # noqa: F401
_merge_dictionaries, # noqa: F401
get_features_as_tensor_dict, # noqa: F401
get_keras_parse_fn, # noqa: F401
get_serving_input_receiver_fn_feature_dict, # noqa: F401
get_string_tensor_parse_fn, # noqa: F401
get_string_tensor_serving_input_receiver_fn, # noqa: F401
get_supervised_input_receiver_fn_feature_dict, # noqa: F401
parse_string_tensor, # noqa: F401
)