Index

_

__init__() (merlin.models.tf.AsDenseFeatures method) (merlin.models.tf.AsSparseFeatures method) (merlin.models.tf.AsTabular method) (merlin.models.tf.AvgPrecisionAt method) (merlin.models.tf.BinaryClassificationTask method) (merlin.models.tf.CachedCrossBatchSampler method) (merlin.models.tf.CachedUniformSampler method) (merlin.models.tf.CausalLanguageModeling method) (merlin.models.tf.CGCBlock method) (merlin.models.tf.ContinuousFeatures method) (merlin.models.tf.dataset.BatchedDataset method) (merlin.models.tf.DotProductInteraction method) (merlin.models.tf.DualEncoderBlock method) (merlin.models.tf.EmbeddingFeatures method) (merlin.models.tf.ExpandDims method) (merlin.models.tf.Filter method) (merlin.models.tf.InBatchSampler method) (merlin.models.tf.ItemRetrievalTask method) (merlin.models.tf.ItemSampler method) (merlin.models.tf.losses.BPRLoss method) (merlin.models.tf.losses.BPRmaxLoss method) (merlin.models.tf.losses.CategoricalCrossEntropy method) (merlin.models.tf.losses.HingeLoss method) (merlin.models.tf.losses.LogisticLoss method) (merlin.models.tf.losses.SparseCategoricalCrossEntropy method) (merlin.models.tf.losses.TOP1Loss method) (merlin.models.tf.losses.TOP1maxLoss method) (merlin.models.tf.losses.TOP1v2Loss method) (merlin.models.tf.MaskedLanguageModeling method) (merlin.models.tf.MultiClassClassificationTask method) (merlin.models.tf.NDCGAt method) (merlin.models.tf.ParallelBlock method) (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PopularityBasedSampler method) (merlin.models.tf.PredictionTask method) (merlin.models.tf.RecallAt method) (merlin.models.tf.RegressionTask method) (merlin.models.tf.ResidualBlock method) (merlin.models.tf.SequenceEmbeddingFeatures method) (merlin.models.tf.SequentialBlock method) (merlin.models.tf.StochasticSwapNoise method) (merlin.models.tf.TabularBlock method) (merlin.models.tf.TwoTowerBlock method) (merlin.models.utils.misc_utils.Timing method) (merlin.models.utils.registry.Registry method) (merlin.models.utils.registry.RegistryMixin method)

A

add() (merlin.models.tf.CachedCrossBatchSampler method) (merlin.models.tf.CachedUniformSampler method) (merlin.models.tf.InBatchSampler method) (merlin.models.tf.ItemSampler method) (merlin.models.tf.PopularityBasedSampler method) add_branch() (merlin.models.tf.ParallelBlock method) add_task() (merlin.models.tf.ParallelPredictionBlock method) aggregation() (merlin.models.tf.TabularBlock property)

apply_mask_to_inputs() (merlin.models.tf.CausalLanguageModeling method) apply_to_all() (merlin.models.tf.TabularBlock method) apply_to_branch() (merlin.models.tf.ParallelBlock method) AsDenseFeatures (class in merlin.models.tf) AsSparseFeatures (class in merlin.models.tf) AsTabular (class in merlin.models.tf) attach_metrics_calculation_to_loss() (merlin.models.tf.PredictionTask method) augment() (merlin.models.tf.StochasticSwapNoise method) AvgPrecisionAt (class in merlin.models.tf)

B

batch_size() (merlin.models.tf.InBatchSampler property) BatchedDataset (class in merlin.models.tf.dataset) BinaryClassificationTask (class in merlin.models.tf) BPRLoss (class in merlin.models.tf.losses) BPRmaxLoss (class in merlin.models.tf.losses) build() (merlin.models.tf.CachedCrossBatchSampler method) (merlin.models.tf.DotProductInteraction method) (merlin.models.tf.EmbeddingFeatures method) (merlin.models.tf.InBatchSampler method) (merlin.models.tf.ParallelBlock method) (merlin.models.tf.SequentialBlock method) (merlin.models.tf.TabularBlock method)

build_task() (merlin.models.tf.PredictionTask method)

C

CachedCrossBatchSampler (class in merlin.models.tf) CachedUniformSampler (class in merlin.models.tf) calculate_batch_size_from_input_shapes() (merlin.models.tf.TabularBlock static method) calculate_metrics() (merlin.models.tf.PredictionTask method) call() (merlin.models.tf.AsDenseFeatures method) (merlin.models.tf.AsSparseFeatures method) (merlin.models.tf.AsTabular method) (merlin.models.tf.BinaryClassificationTask method) (merlin.models.tf.CachedCrossBatchSampler method) (merlin.models.tf.CGCBlock method) (merlin.models.tf.ContinuousFeatures method) (merlin.models.tf.DotProductInteraction method) (merlin.models.tf.EmbeddingFeatures method) (merlin.models.tf.ExpandDims method) (merlin.models.tf.Filter method) (merlin.models.tf.InBatchSampler method) (merlin.models.tf.MultiClassClassificationTask method) (merlin.models.tf.ParallelBlock method) (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PopularityBasedSampler method) (merlin.models.tf.RegressionTask method) (merlin.models.tf.SequentialBlock method) (merlin.models.tf.StochasticSwapNoise method) (merlin.models.tf.TabularBlock method) call_outputs() (merlin.models.tf.SequentialBlock method) camelcase_to_snakecase() (in module merlin.models.utils.registry) categorical_cardinalities() (in module merlin.models.utils.schema_utils) categorical_domains() (in module merlin.models.utils.schema_utils) CategoricalCrossEntropy (class in merlin.models.tf.losses) CausalLanguageModeling (class in merlin.models.tf) CGCBlock (class in merlin.models.tf) check_feature() (merlin.models.tf.Filter method) child_name() (merlin.models.tf.PredictionTask method) class_registry() (merlin.models.utils.registry.Registry class method) compute() (merlin.models.tf.losses.BPRLoss method) (merlin.models.tf.losses.BPRmaxLoss method) (merlin.models.tf.losses.HingeLoss method) (merlin.models.tf.losses.LogisticLoss method) (merlin.models.tf.losses.TOP1Loss method) (merlin.models.tf.losses.TOP1maxLoss method) (merlin.models.tf.losses.TOP1v2Loss method)

compute_call_output_shape() (merlin.models.tf.CGCBlock method) (merlin.models.tf.ContinuousFeatures method) (merlin.models.tf.EmbeddingFeatures method) (merlin.models.tf.Filter method) (merlin.models.tf.ParallelBlock method) (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.SequenceEmbeddingFeatures method) (merlin.models.tf.TabularBlock method) compute_loss() (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PredictionTask method) (merlin.models.tf.SequentialBlock method) compute_mask() (merlin.models.tf.SequenceEmbeddingFeatures method) compute_mask_schema() (merlin.models.tf.CausalLanguageModeling method) (merlin.models.tf.MaskedLanguageModeling method) compute_output_shape() (merlin.models.tf.AsDenseFeatures method) (merlin.models.tf.AsSparseFeatures method) (merlin.models.tf.AsTabular method) (merlin.models.tf.BinaryClassificationTask method) (merlin.models.tf.DotProductInteraction method) (merlin.models.tf.ExpandDims method) (merlin.models.tf.SequentialBlock method) (merlin.models.tf.StochasticSwapNoise method) (merlin.models.tf.TabularBlock method) compute_output_signature() (merlin.models.tf.SequentialBlock method) configure_tensorflow() (in module merlin.models.loader.tf_utils) ContinuousEmbedding() (in module merlin.models.tf) ContinuousFeatures (class in merlin.models.tf) create_categorical_column() (in module merlin.models.utils.schema_utils) create_continuous_column() (in module merlin.models.utils.schema_utils) CrossBlock() (in module merlin.models.tf)

D

DCNModel() (in module merlin.models.tf) default_key() (merlin.models.utils.registry.Registry method) DEFAULT_LOSS (merlin.models.tf.BinaryClassificationTask attribute) (merlin.models.tf.ItemRetrievalTask attribute) (merlin.models.tf.MultiClassClassificationTask attribute) (merlin.models.tf.RegressionTask attribute) DEFAULT_METRICS (merlin.models.tf.BinaryClassificationTask attribute) (merlin.models.tf.ItemRetrievalTask attribute) (merlin.models.tf.MultiClassClassificationTask attribute) (merlin.models.tf.RegressionTask attribute)

default_name() (in module merlin.models.utils.registry) default_object_name() (in module merlin.models.utils.registry) DenseResidualBlock() (in module merlin.models.tf) device_mem_size() (in module merlin.models.loader.utils) display_list_by_prefix() (in module merlin.models.utils.registry) DLRMBlock() (in module merlin.models.tf) DLRMModel() (in module merlin.models.tf) DotProductInteraction (class in merlin.models.tf) DualEncoderBlock (class in merlin.models.tf)

E

embedding_table_dataset() (merlin.models.tf.EmbeddingFeatures method) embedding_table_df() (merlin.models.tf.EmbeddingFeatures method)

EmbeddingFeatures (class in merlin.models.tf) ExpandDims (class in merlin.models.tf) export_embedding_table() (merlin.models.tf.EmbeddingFeatures method)

F

Filter (class in merlin.models.tf) filter_dict_by_schema() (in module merlin.models.utils.schema_utils) filter_features() (merlin.models.tf.SequentialBlock property) filter_kwargs() (in module merlin.models.utils.misc_utils) first() (merlin.models.tf.SequentialBlock property) from_config() (merlin.models.tf.BinaryClassificationTask class method) (merlin.models.tf.DualEncoderBlock class method) (merlin.models.tf.EmbeddingFeatures class method) (merlin.models.tf.MultiClassClassificationTask class method) (merlin.models.tf.ParallelBlock class method) (merlin.models.tf.ParallelPredictionBlock class method) (merlin.models.tf.PredictionTask class method) (merlin.models.tf.RegressionTask class method) (merlin.models.tf.SequentialBlock class method) (merlin.models.tf.TabularBlock class method)

from_features() (merlin.models.tf.ContinuousFeatures class method) (merlin.models.tf.TabularBlock class method) from_schema() (merlin.models.tf.EmbeddingFeatures class method) (merlin.models.tf.MultiClassClassificationTask class method) (merlin.models.tf.ParallelPredictionBlock class method) (merlin.models.tf.TabularBlock class method)

G

get() (merlin.models.utils.registry.Registry method) get_config() (merlin.models.tf.AsDenseFeatures method) (merlin.models.tf.AsTabular method) (merlin.models.tf.BinaryClassificationTask method) (merlin.models.tf.ContinuousFeatures method) (merlin.models.tf.DotProductInteraction method) (merlin.models.tf.EmbeddingFeatures method) (merlin.models.tf.Filter method) (merlin.models.tf.MaskedLanguageModeling method) (merlin.models.tf.MultiClassClassificationTask method) (merlin.models.tf.ParallelBlock method) (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PredictionTask method) (merlin.models.tf.RegressionTask method) (merlin.models.tf.SequenceEmbeddingFeatures method) (merlin.models.tf.SequentialBlock method) (merlin.models.tf.StochasticSwapNoise method) (merlin.models.tf.TabularBlock method)

get_dataset_schema_from_feature_columns() (in module merlin.models.loader.tf_utils) get_embedding_size_from_cardinality() (in module merlin.models.utils.schema_utils) get_embedding_sizes_from_schema() (in module merlin.models.utils.schema_utils) get_filenames() (in module merlin.models.utils.misc_utils) get_label_feature_name() (in module merlin.models.utils.misc_utils) get_object_size() (in module merlin.models.utils.misc_utils) get_parquet_files_names() (in module merlin.models.utils.misc_utils) get_tasks_from_schema() (merlin.models.tf.ParallelPredictionBlock class method) get_timestamp_feature_name() (in module merlin.models.utils.misc_utils)

H

HingeLoss (class in merlin.models.tf.losses)

I

InBatchSampler (class in merlin.models.tf) InputBlock() (in module merlin.models.tf) inputs() (merlin.models.tf.SequentialBlock property) is_input() (merlin.models.tf.TabularBlock property) is_tabular() (merlin.models.tf.AsTabular property) (merlin.models.tf.SequentialBlock property) (merlin.models.tf.TabularBlock property)

item_block() (merlin.models.tf.DualEncoderBlock method) item_embeddings_queue() (merlin.models.tf.CachedCrossBatchSampler property) ItemRetrievalTask (class in merlin.models.tf) items() (merlin.models.utils.registry.Registry method) ItemSampler (class in merlin.models.tf)

K

keys() (merlin.models.utils.registry.Registry method)

L

last() (merlin.models.tf.SequentialBlock property) LogisticLoss (class in merlin.models.tf.losses)

lookup_feature() (merlin.models.tf.EmbeddingFeatures method) (merlin.models.tf.SequenceEmbeddingFeatures method) losses() (merlin.models.tf.SequentialBlock property)

M

map() (merlin.models.tf.dataset.BatchedDataset method) MaskedLanguageModeling (class in merlin.models.tf) MatrixFactorizationBlock() (in module merlin.models.tf) MatrixFactorizationModel() (in module merlin.models.tf) max_num_samples() (merlin.models.tf.ItemSampler property) metric_result_dict() (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PredictionTask method)

metric_results() (merlin.models.tf.MultiClassClassificationTask method) (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PredictionTask method) metrics() (merlin.models.tf.ParallelPredictionBlock property) MLPBlock() (in module merlin.models.tf) MMOEBlock() (in module merlin.models.tf) MultiClassClassificationTask (class in merlin.models.tf)

N

name() (merlin.models.utils.registry.Registry property) NDCGAt (class in merlin.models.tf)

NextItemPredictionTask() (in module merlin.models.tf) non_trainable_weights() (merlin.models.tf.SequentialBlock property)

O

on_set() (merlin.models.utils.registry.Registry method)

P

parallel_dict() (merlin.models.tf.ParallelBlock property) parallel_layers (merlin.models.tf.ParallelPredictionBlock attribute) (merlin.models.tf.ResidualBlock attribute) (merlin.models.tf.TwoTowerBlock attribute) parallel_values() (merlin.models.tf.ParallelBlock property) ParallelBlock (class in merlin.models.tf) ParallelPredictionBlock (class in merlin.models.tf) parse() (merlin.models.utils.registry.Registry method) (merlin.models.utils.registry.RegistryMixin class method) parse_config() (merlin.models.tf.ParallelBlock class method)

pop_labels() (merlin.models.tf.ParallelPredictionBlock method) PopularityBasedSampler (class in merlin.models.tf) post() (merlin.models.tf.TabularBlock property) post_call() (merlin.models.tf.TabularBlock method) pre() (merlin.models.tf.TabularBlock property) pre_call() (merlin.models.tf.PredictionTask method) (merlin.models.tf.TabularBlock method) pre_eval_topk() (merlin.models.tf.PredictionTask property) pre_loss() (merlin.models.tf.PredictionTask method) PredictionTask (class in merlin.models.tf) PredictionTasks() (in module merlin.models.tf)

Q

query_block() (merlin.models.tf.DualEncoderBlock method)

R

ranking_metrics() (in module merlin.models.tf) RecallAt (class in merlin.models.tf) register() (merlin.models.utils.registry.Registry method) register_with_multiple_names() (merlin.models.utils.registry.Registry method) Registry (class in merlin.models.utils.registry) registry (merlin.models.utils.registry.RegistryMixin attribute) RegistryMixin (class in merlin.models.utils.registry) RegressionTask (class in merlin.models.tf) regularizers() (merlin.models.tf.SequentialBlock property) repr_add() (merlin.models.tf.PredictionTask method) (merlin.models.tf.TabularBlock method)

repr_extra() (merlin.models.tf.ContinuousFeatures method) (merlin.models.tf.TabularBlock method) repr_ignore() (merlin.models.tf.ContinuousFeatures method) (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.TabularBlock method) required_features() (merlin.models.tf.ItemSampler property) reset_metrics() (merlin.models.tf.ParallelPredictionBlock method) (merlin.models.tf.PredictionTask method) ResidualBlock (class in merlin.models.tf) retrieval_scorer() (merlin.models.tf.ItemRetrievalTask property)

S

safe_json() (in module merlin.models.utils.misc_utils) sample() (merlin.models.tf.CachedCrossBatchSampler method) (merlin.models.tf.InBatchSampler method) (merlin.models.tf.ItemSampler method) (merlin.models.tf.PopularityBasedSampler method) schema_to_tensorflow_metadata_json() (in module merlin.models.utils.schema_utils) select_by_name() (merlin.models.tf.ParallelBlock method) select_targets() (in module merlin.models.utils.schema_utils) SequenceEmbeddingFeatures (class in merlin.models.tf) SequentialBlock (class in merlin.models.tf) set_aggregation() (merlin.models.tf.TabularBlock method)

set_batch_size() (merlin.models.tf.InBatchSampler method) set_max_num_samples() (merlin.models.tf.ItemSampler method) set_post() (merlin.models.tf.TabularBlock method) set_pre() (merlin.models.tf.TabularBlock method) set_retrieval_cache_query() (merlin.models.tf.ItemRetrievalTask method) set_schema() (merlin.models.tf.Filter method) (merlin.models.tf.SequentialBlock method) (merlin.models.tf.TabularBlock method) snakecase_to_camelcase() (in module merlin.models.utils.registry) SparseCategoricalCrossEntropy (class in merlin.models.tf.losses) StochasticSwapNoise (class in merlin.models.tf) super() (merlin.models.tf.TabularBlock method)

T

table_config() (merlin.models.tf.EmbeddingFeatures method) TabularBlock (class in merlin.models.tf) task_blocks() (merlin.models.tf.ParallelPredictionBlock property) task_name() (merlin.models.tf.PredictionTask property) task_names() (merlin.models.tf.ParallelPredictionBlock property) task_names_from_schema() (merlin.models.tf.ParallelPredictionBlock class method) tensorflow_metadata_json_to_schema() (in module merlin.models.utils.schema_utils)

Timing (class in merlin.models.utils.misc_utils) TOP1Loss (class in merlin.models.tf.losses) TOP1maxLoss (class in merlin.models.tf.losses) TOP1v2Loss (class in merlin.models.tf.losses) trainable() (merlin.models.tf.SequentialBlock property) trainable_weights() (merlin.models.tf.SequentialBlock property) TwoTowerBlock (class in merlin.models.tf) TwoTowerModel() (in module merlin.models.tf)

V

validate() (merlin.models.utils.registry.Registry method)

validate_dataset() (in module merlin.models.utils.misc_utils) values() (merlin.models.utils.registry.Registry method)

Y

YoutubeDNNRetrievalModel() (in module merlin.models.tf)