mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-26 13:06:05 +02:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
18 lines
729 B
Python
18 lines
729 B
Python
'''
|
|
Contains implemenations of DataRecordTrainer.get_export_output_fns that specify how to
|
|
export model graph outputs from build_graph to DataRecords for prediction servers.
|
|
|
|
Modelers can use the functions in this module as the export_output_fn parameter of
|
|
the DataRecordTrainer constructor to customize how to export their model outputs.
|
|
|
|
Modelers may also provide a custom implementation of export_output_fn using these as reference.
|
|
'''
|
|
|
|
# pylint: disable=invalid-name
|
|
from twitter.deepbird.io.legacy.export_output_fns import (
|
|
batch_prediction_continuous_output_fn, # noqa: F401
|
|
batch_prediction_tensor_output_fn, # noqa: F401
|
|
default_output_fn, # noqa: F401
|
|
variable_length_continuous_output_fn, # noqa: F401
|
|
)
|