View source on GitHub
|
Ranking pipeline to train tf.keras.Model in tfr.keras.
Classes
class AbstractDatasetBuilder: Interface for datasets and signatures.
class AbstractPipeline: Interface for ranking pipeline to train a tf.keras.Model.
class BaseDatasetBuilder: Builds datasets from feature specs.
class DatasetHparams: Hyperparameters used in BaseDatasetBuilder.
class ModelFitPipeline: Pipeline using model.fit to train a ranking tf.keras.Model.
class MultiLabelDatasetBuilder: Builds datasets for multi-task training.
class MultiTaskPipeline: Pipeline for multi-task training.
class NullDatasetBuilder: A no-op wrapper of datasets and signatures.
class PipelineHparams: Hyperparameters used in ModelFitPipeline.
class SimpleDatasetBuilder: Builds datasets from feature specs with a single label spec.
class SimplePipeline: Pipleine for single-task training.