Transformers4Rec
Contents
Introduction
Why Transformers4Rec?
Model Architectures
Training and Evaluation
End-to-End Pipeline with Hugging Face Transformers and NVIDIA Merlin
Multi-GPU data-parallel training using the
Trainer
class
Example Notebooks
API Documentation
Additional Resources
Transformers4Rec
»
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
A
add() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
add_module() (transformers4rec.torch.block.base.SequentialBlock method)
(transformers4rec.torch.SequentialBlock method)
add_module_and_maybe_build() (transformers4rec.torch.block.base.SequentialBlock method)
(transformers4rec.torch.SequentialBlock method)
aggregation() (transformers4rec.torch.TabularModule property)
AlbertConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
ALPHA (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
Annotation (class in merlin_standard_lib.proto.schema_bp)
annotation (merlin_standard_lib.proto.schema_bp.Feature attribute)
append() (transformers4rec.torch.SequentialTabularTransformations method)
apply() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
apply_inverse() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
apply_mask_to_inputs() (transformers4rec.torch.masking.CausalLanguageModeling method)
(transformers4rec.torch.masking.MaskedLanguageModeling method)
(transformers4rec.torch.masking.MaskSequence method)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking.CausalLanguageModeling method)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking.MaskedLanguageModeling method)
as_tabular() (transformers4rec.torch.block.base.BlockBase method)
(transformers4rec.torch.block.base.SequentialBlock method)
(transformers4rec.torch.BlockBase method)
(transformers4rec.torch.SequentialBlock method)
AsTabular (class in transformers4rec.torch)
atleast_1d() (in module transformers4rec.torch.utils.torch_utils)
augment() (transformers4rec.torch.StochasticSwapNoise method)
(transformers4rec.torch.tabular.transformations.StochasticSwapNoise method)
AvgPrecisionAt (class in transformers4rec.torch.ranking_metric)
B
batch_size (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
BertConfig (class in transformers4rec.config.transformer)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
BETA (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
BinaryClassificationPrepareBlock (class in transformers4rec.torch.model.prediction_task)
BinaryClassificationTask (class in transformers4rec.torch)
(class in transformers4rec.torch.model.prediction_task)
Block (class in transformers4rec.torch)
(class in transformers4rec.torch.block.base)
BlockBase (class in transformers4rec.torch)
(class in transformers4rec.torch.block.base)
bool_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
BoolDomain (class in merlin_standard_lib.proto.schema_bp)
build() (transformers4rec.config.transformer.AlbertConfig class method)
(transformers4rec.config.transformer.BertConfig class method)
(transformers4rec.config.transformer.ElectraConfig class method)
(transformers4rec.config.transformer.GPT2Config class method)
(transformers4rec.config.transformer.LongformerConfig class method)
(transformers4rec.config.transformer.ReformerConfig class method)
(transformers4rec.config.transformer.RobertaConfig class method)
(transformers4rec.config.transformer.T4RecConfig class method)
(transformers4rec.config.transformer.TransfoXLConfig class method)
(transformers4rec.config.transformer.XLNetConfig class method)
(transformers4rec.torch.AlbertConfig class method)
(transformers4rec.torch.block.base.BuildableBlock method)
(transformers4rec.torch.block.base.SequentialBlock method)
(transformers4rec.torch.block.mlp.MLPBlock method)
(transformers4rec.torch.ElectraConfig class method)
(transformers4rec.torch.features.embedding.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.GPT2Config class method)
(transformers4rec.torch.Head method)
(transformers4rec.torch.LongformerConfig class method)
(transformers4rec.torch.MergeTabular method)
(transformers4rec.torch.MLPBlock method)
(transformers4rec.torch.model.prediction_task.BinaryClassificationPrepareBlock method)
(transformers4rec.torch.model.prediction_task.NextItemPredictionPrepareBlock method)
(transformers4rec.torch.model.prediction_task.NextItemPredictionTask method)
(transformers4rec.torch.model.prediction_task.RegressionPrepareBlock method)
(transformers4rec.torch.NextItemPredictionTask method)
(transformers4rec.torch.PredictionTask method)
(transformers4rec.torch.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.ReformerConfig class method)
(transformers4rec.torch.SequentialBlock method)
(transformers4rec.torch.T4RecConfig class method)
(transformers4rec.torch.tabular.transformations.TabularLayerNorm method)
(transformers4rec.torch.TabularBlock method)
(transformers4rec.torch.TabularLayerNorm method)
(transformers4rec.torch.TransfoXLConfig class method)
(transformers4rec.torch.utils.torch_utils.OutputSizeMixin method)
(transformers4rec.torch.XLNetConfig class method)
build_blocks() (in module transformers4rec.torch)
(in module transformers4rec.torch.block.base)
BuildableBlock (class in transformers4rec.torch.block.base)
BYTES (merlin_standard_lib.proto.schema_bp.FeatureType attribute)
bytes_value (merlin_standard_lib.proto.schema_bp.TensorRepresentationDefaultValue attribute)
C
calculate_batch_size_from_input_size() (in module transformers4rec.torch.utils.torch_utils)
calculate_metrics() (transformers4rec.torch.Head method)
(transformers4rec.torch.Model method)
(transformers4rec.torch.model.prediction_task.NextItemPredictionTask method)
(transformers4rec.torch.NextItemPredictionTask method)
(transformers4rec.torch.PredictionTask method)
(transformers4rec.torch.utils.torch_utils.MetricsMixin method)
categorical_cardinalities() (in module merlin_standard_lib)
(in module merlin_standard_lib.schema.schema)
categorical_module() (transformers4rec.torch.features.tabular.TabularFeatures property)
(transformers4rec.torch.TabularFeatures property)
CausalLanguageModeling (class in transformers4rec.torch.masking)
check_gpu() (in module transformers4rec.torch.utils.torch_utils)
check_inputs() (in module transformers4rec.torch.utils.torch_utils)
check_schema() (transformers4rec.config.schema.SchemaMixin method)
child_name() (transformers4rec.torch.PredictionTask method)
column_name (merlin_standard_lib.proto.schema_bp.TensorRepresentationDenseTensor attribute)
(merlin_standard_lib.proto.schema_bp.TensorRepresentationVarLenSparseTensor attribute)
column_names() (merlin_standard_lib.Schema property)
(merlin_standard_lib.schema.schema.Schema property)
(transformers4rec.torch.Schema property)
column_schemas() (merlin_standard_lib.Schema property)
(merlin_standard_lib.schema.schema.Schema property)
(transformers4rec.torch.Schema property)
ColumnSchema (class in merlin_standard_lib)
(class in merlin_standard_lib.schema.schema)
combiner (transformers4rec.torch.features.embedding.TableConfig attribute)
(transformers4rec.torch.TableConfig attribute)
comment (merlin_standard_lib.proto.schema_bp.Annotation attribute)
compute() (transformers4rec.torch.ranking_metric.RankingMetric method)
compute_loss() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
(transformers4rec.torch.utils.torch_utils.LossMixin method)
compute_masked_targets() (transformers4rec.torch.masking.MaskSequence method)
(transformers4rec.torch.masking.PermutationLanguageModeling method)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking.PermutationLanguageModeling method)
compute_metrics() (transformers4rec.torch.Head method)
(transformers4rec.torch.Model method)
(transformers4rec.torch.model.prediction_task.NextItemPredictionTask method)
(transformers4rec.torch.NextItemPredictionTask method)
(transformers4rec.torch.PredictionTask method)
(transformers4rec.torch.utils.torch_utils.MetricsMixin method)
compute_metrics_each_n_steps (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
ConcatFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.aggregation)
continuous_module() (transformers4rec.torch.features.tabular.TabularFeatures property)
(transformers4rec.torch.TabularFeatures property)
CONTINUOUS_MODULE_CLASS (transformers4rec.torch.features.tabular.TabularFeatures attribute)
(transformers4rec.torch.TabularFeatures attribute)
ContinuousFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.continuous)
ConvBertConfig (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
copy() (merlin_standard_lib.ColumnSchema method)
(merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.ColumnSchema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
coverage (merlin_standard_lib.proto.schema_bp.NaturalLanguageDomain attribute)
create() (merlin_standard_lib.Schema class method)
(merlin_standard_lib.schema.schema.Schema class method)
(transformers4rec.torch.Schema class method)
create_categorical() (merlin_standard_lib.ColumnSchema class method)
(merlin_standard_lib.schema.schema.ColumnSchema class method)
create_continuous() (merlin_standard_lib.ColumnSchema class method)
(merlin_standard_lib.schema.schema.ColumnSchema class method)
create_output_placeholder() (in module transformers4rec.torch.utils.torch_utils)
create_scheduler() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
D
data_loader_engine (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
dataset (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
DatasetConstraints (class in merlin_standard_lib.proto.schema_bp)
DatasetMock (class in transformers4rec.torch.trainer)
DCGAt (class in transformers4rec.torch.ranking_metric)
DebertaConfig (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
DEBUG_ONLY (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
DEFAULT_LOSS (transformers4rec.torch.BinaryClassificationTask attribute)
(transformers4rec.torch.model.prediction_task.BinaryClassificationTask attribute)
(transformers4rec.torch.model.prediction_task.RegressionTask attribute)
(transformers4rec.torch.RegressionTask attribute)
DEFAULT_MASKING (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
DEFAULT_METRICS (transformers4rec.torch.BinaryClassificationTask attribute)
(transformers4rec.torch.model.prediction_task.BinaryClassificationTask attribute)
(transformers4rec.torch.model.prediction_task.NextItemPredictionTask attribute)
(transformers4rec.torch.model.prediction_task.RegressionTask attribute)
(transformers4rec.torch.NextItemPredictionTask attribute)
(transformers4rec.torch.RegressionTask attribute)
default_value (merlin_standard_lib.proto.schema_bp.TensorRepresentationDenseTensor attribute)
dense_shape (merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
(merlin_standard_lib.proto.schema_bp.TensorRepresentationSparseTensor attribute)
dense_tensor (merlin_standard_lib.proto.schema_bp.TensorRepresentation attribute)
DenseBlock (class in transformers4rec.torch.block.mlp)
deprecated (merlin_standard_lib.proto.schema_bp.Feature attribute)
DEPRECATED (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
deprecated (merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
device() (transformers4rec.torch.utils.data_utils.DLDataLoader property)
dim (merlin_standard_lib.proto.schema_bp.FixedShape attribute)
(transformers4rec.torch.features.embedding.TableConfig attribute)
(transformers4rec.torch.TableConfig attribute)
DISABLED (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
disallow_inf (merlin_standard_lib.proto.schema_bp.FloatDomain attribute)
disallow_nan (merlin_standard_lib.proto.schema_bp.FloatDomain attribute)
DistilBertConfig (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
distribution_constraints (merlin_standard_lib.proto.schema_bp.Feature attribute)
DistributionConstraints (class in merlin_standard_lib.proto.schema_bp)
DLDataLoader (class in transformers4rec.torch.utils.data_utils)
domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
drift_comparator (merlin_standard_lib.proto.schema_bp.Feature attribute)
drop_last (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
E
ElectraConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
ElementwiseFeatureAggregation (class in transformers4rec.torch.tabular.aggregation)
ElementwiseSum (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.aggregation)
ElementwiseSumItemMulti (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.aggregation)
embedding_dim (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
EMBEDDING_MODULE_CLASS (transformers4rec.torch.features.sequence.TabularSequenceFeatures attribute)
(transformers4rec.torch.features.tabular.TabularFeatures attribute)
(transformers4rec.torch.TabularFeatures attribute)
(transformers4rec.torch.TabularSequenceFeatures attribute)
EmbeddingBagWrapper (class in transformers4rec.torch.features.embedding)
EmbeddingFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
eval_on_test_set (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
eval_steps_on_train_set (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
evaluate() (transformers4rec.torch.Model method)
evaluation_loop() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
excluded_int_tokens (merlin_standard_lib.proto.schema_bp.FeatureCoverageConstraints attribute)
excluded_int_value (merlin_standard_lib.proto.schema_bp.SequenceLengthConstraints attribute)
excluded_string_tokens (merlin_standard_lib.proto.schema_bp.FeatureCoverageConstraints attribute)
excluded_string_value (merlin_standard_lib.proto.schema_bp.SequenceLengthConstraints attribute)
experiments_group (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
extra_metadata (merlin_standard_lib.proto.schema_bp.Annotation attribute)
extract_topk() (in module transformers4rec.torch.utils.torch_utils)
F
false_value (merlin_standard_lib.proto.schema_bp.BoolDomain attribute)
Feature (class in merlin_standard_lib.proto.schema_bp)
feature (merlin_standard_lib.proto.schema_bp.StructDomain attribute)
(merlin_standard_lib.proto.schema_bp.WeightedFeature attribute)
(merlin_standard_lib.Schema attribute)
(merlin_standard_lib.schema.schema.Schema attribute)
(transformers4rec.torch.Schema attribute)
feature_path (merlin_standard_lib.proto.schema_bp.TensorRepresentationRaggedTensor attribute)
FeatureComparator (class in merlin_standard_lib.proto.schema_bp)
FeatureConfig (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
FeatureCoverageConstraints (class in merlin_standard_lib.proto.schema_bp)
FeaturePresence (class in merlin_standard_lib.proto.schema_bp)
FeaturePresenceWithinGroup (class in merlin_standard_lib.proto.schema_bp)
FeatureType (class in merlin_standard_lib.proto.schema_bp)
filter_column_schemas() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
filter_columns_from_dict() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
FilterFeatures (class in transformers4rec.torch)
fit() (transformers4rec.torch.Model method)
fit_and_evaluate() (in module transformers4rec.torch.utils.examples_utils)
FixedShape (class in merlin_standard_lib.proto.schema_bp)
FixedShapeDim (class in merlin_standard_lib.proto.schema_bp)
FLOAT (merlin_standard_lib.proto.schema_bp.FeatureType attribute)
float_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
float_value (merlin_standard_lib.proto.schema_bp.TensorRepresentationDefaultValue attribute)
FloatDomain (class in merlin_standard_lib.proto.schema_bp)
FORMAT_UNKNOWN (merlin_standard_lib.proto.schema_bp.TimeDomainIntegerTimeFormat attribute)
(merlin_standard_lib.proto.schema_bp.TimeOfDayDomainIntegerTimeOfDayFormat attribute)
forward() (transformers4rec.torch.AsTabular method)
(transformers4rec.torch.Block method)
(transformers4rec.torch.block.base.Block method)
(transformers4rec.torch.block.base.SequentialBlock method)
(transformers4rec.torch.block.transformer.GPT2Prepare method)
(transformers4rec.torch.block.transformer.TransformerBlock method)
(transformers4rec.torch.block.transformer.TransformerPrepare method)
(transformers4rec.torch.ConcatFeatures method)
(transformers4rec.torch.ContinuousFeatures method)
(transformers4rec.torch.ElementwiseSum method)
(transformers4rec.torch.ElementwiseSumItemMulti method)
(transformers4rec.torch.EmbeddingFeatures method)
(transformers4rec.torch.features.continuous.ContinuousFeatures method)
(transformers4rec.torch.features.embedding.EmbeddingBagWrapper method)
(transformers4rec.torch.features.embedding.EmbeddingFeatures method)
(transformers4rec.torch.features.embedding.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.features.embedding.PretrainedEmbeddingsInitializer method)
(transformers4rec.torch.features.embedding.SoftEmbedding method)
(transformers4rec.torch.features.sequence.TabularSequenceFeatures method)
(transformers4rec.torch.FilterFeatures method)
(transformers4rec.torch.Head method)
(transformers4rec.torch.masking.MaskSequence method)
(transformers4rec.torch.MergeTabular method)
(transformers4rec.torch.Model method)
(transformers4rec.torch.model.prediction_task.LogUniformSampler method)
(transformers4rec.torch.model.prediction_task.NextItemPredictionTask method)
(transformers4rec.torch.NextItemPredictionTask method)
(transformers4rec.torch.PredictionTask method)
(transformers4rec.torch.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.PretrainedEmbeddingsInitializer method)
(transformers4rec.torch.SequentialBlock method)
(transformers4rec.torch.SoftEmbedding method)
(transformers4rec.torch.StackFeatures method)
(transformers4rec.torch.StochasticSwapNoise method)
(transformers4rec.torch.tabular.aggregation.ConcatFeatures method)
(transformers4rec.torch.tabular.aggregation.ElementwiseSum method)
(transformers4rec.torch.tabular.aggregation.ElementwiseSumItemMulti method)
(transformers4rec.torch.tabular.aggregation.StackFeatures method)
(transformers4rec.torch.tabular.transformations.StochasticSwapNoise method)
(transformers4rec.torch.tabular.transformations.TabularDropout method)
(transformers4rec.torch.tabular.transformations.TabularLayerNorm method)
(transformers4rec.torch.TabularAggregation method)
(transformers4rec.torch.TabularDropout method)
(transformers4rec.torch.TabularLayerNorm method)
(transformers4rec.torch.TabularModule method)
(transformers4rec.torch.TabularSequenceFeatures method)
(transformers4rec.torch.TabularTransformation method)
(transformers4rec.torch.TransformerBlock method)
(transformers4rec.torch.utils.torch_utils.LambdaModule method)
forward_output_size() (transformers4rec.torch.AsTabular method)
(transformers4rec.torch.Block method)
(transformers4rec.torch.block.base.Block method)
(transformers4rec.torch.block.base.SequentialBlock method)
(transformers4rec.torch.block.mlp.DenseBlock method)
(transformers4rec.torch.block.transformer.TransformerBlock method)
(transformers4rec.torch.ConcatFeatures method)
(transformers4rec.torch.ContinuousFeatures method)
(transformers4rec.torch.ElementwiseSum method)
(transformers4rec.torch.ElementwiseSumItemMulti method)
(transformers4rec.torch.EmbeddingFeatures method)
(transformers4rec.torch.features.continuous.ContinuousFeatures method)
(transformers4rec.torch.features.embedding.EmbeddingFeatures method)
(transformers4rec.torch.features.embedding.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.features.sequence.SequenceEmbeddingFeatures method)
(transformers4rec.torch.features.sequence.TabularSequenceFeatures method)
(transformers4rec.torch.features.tabular.TabularFeatures method)
(transformers4rec.torch.FilterFeatures method)
(transformers4rec.torch.masking.MaskSequence method)
(transformers4rec.torch.MergeTabular method)
(transformers4rec.torch.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.SequenceEmbeddingFeatures method)
(transformers4rec.torch.SequentialBlock method)
(transformers4rec.torch.StackFeatures method)
(transformers4rec.torch.StochasticSwapNoise method)
(transformers4rec.torch.tabular.aggregation.ConcatFeatures method)
(transformers4rec.torch.tabular.aggregation.ElementwiseSum method)
(transformers4rec.torch.tabular.aggregation.ElementwiseSumItemMulti method)
(transformers4rec.torch.tabular.aggregation.StackFeatures method)
(transformers4rec.torch.tabular.transformations.StochasticSwapNoise method)
(transformers4rec.torch.tabular.transformations.TabularDropout method)
(transformers4rec.torch.tabular.transformations.TabularLayerNorm method)
(transformers4rec.torch.TabularDropout method)
(transformers4rec.torch.TabularFeatures method)
(transformers4rec.torch.TabularLayerNorm method)
(transformers4rec.torch.TabularSequenceFeatures method)
(transformers4rec.torch.TransformerBlock method)
(transformers4rec.torch.utils.torch_utils.OutputSizeMixin method)
from_feature_config() (transformers4rec.torch.tabular.transformations.TabularLayerNorm class method)
(transformers4rec.torch.TabularLayerNorm class method)
from_features() (transformers4rec.torch.ContinuousFeatures class method)
(transformers4rec.torch.features.continuous.ContinuousFeatures class method)
(transformers4rec.torch.TabularModule class method)
from_json() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
from_proto_text() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
from_registry() (transformers4rec.torch.block.transformer.TransformerBlock class method)
(transformers4rec.torch.TransformerBlock class method)
from_schema() (transformers4rec.torch.EmbeddingFeatures class method)
(transformers4rec.torch.features.embedding.EmbeddingFeatures class method)
(transformers4rec.torch.features.embedding.PretrainedEmbeddingFeatures class method)
(transformers4rec.torch.features.embedding.SoftEmbeddingFeatures class method)
(transformers4rec.torch.features.sequence.TabularSequenceFeatures class method)
(transformers4rec.torch.features.tabular.TabularFeatures class method)
(transformers4rec.torch.Head class method)
(transformers4rec.torch.PretrainedEmbeddingFeatures class method)
(transformers4rec.torch.SoftEmbeddingFeatures class method)
(transformers4rec.torch.TabularFeatures class method)
(transformers4rec.torch.TabularModule class method)
(transformers4rec.torch.TabularSequenceFeatures class method)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader class method)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader class method)
(transformers4rec.torch.utils.data_utils.T4RecDataLoader class method)
G
get_children_by_class_name() (transformers4rec.torch.block.base.SequentialBlock static method)
(transformers4rec.torch.SequentialBlock static method)
get_embedding_size_from_cardinality() (in module merlin_standard_lib.utils.embedding_utils)
get_embedding_sizes_from_schema() (in module merlin_standard_lib.utils.embedding_utils)
get_eval_dataloader() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
get_fake_tokens() (transformers4rec.torch.masking.ReplacementLanguageModeling method)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking.ReplacementLanguageModeling method)
get_item_ids_from_inputs() (transformers4rec.config.schema.SchemaMixin method)
get_log_uniform_distr() (transformers4rec.torch.model.prediction_task.LogUniformSampler method)
get_output_sizes_from_schema() (in module transformers4rec.torch.utils.torch_utils)
get_padding_mask_from_item_id() (transformers4rec.config.schema.SchemaMixin method)
get_scheduler() (transformers4rec.torch.Trainer static method)
(transformers4rec.torch.trainer.Trainer static method)
get_test_dataloader() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
get_train_dataloader() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
get_unique_sampling_distr() (transformers4rec.torch.model.prediction_task.LogUniformSampler method)
GPT2Config (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
GPT2Prepare (class in transformers4rec.torch.block.transformer)
group_presence (merlin_standard_lib.proto.schema_bp.Feature attribute)
H
Head (class in transformers4rec.torch)
I
image_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
ImageDomain (class in merlin_standard_lib.proto.schema_bp)
in_environment (merlin_standard_lib.proto.schema_bp.Feature attribute)
include_last_offset (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
IncrementalLoggingCallback (class in transformers4rec.torch.trainer)
index_column_names (merlin_standard_lib.proto.schema_bp.TensorRepresentationSparseTensor attribute)
index_feature (merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
infinity_norm (merlin_standard_lib.proto.schema_bp.FeatureComparator attribute)
InfinityNorm (class in merlin_standard_lib.proto.schema_bp)
initializer (transformers4rec.torch.features.embedding.TableConfig attribute)
(transformers4rec.torch.TableConfig attribute)
input_schema() (transformers4rec.torch.Model property)
InputBlock (class in transformers4rec.torch.features.base)
inputs() (transformers4rec.torch.block.base.SequentialBlock property)
(transformers4rec.torch.SequentialBlock property)
INT (merlin_standard_lib.proto.schema_bp.FeatureType attribute)
INT32 (merlin_standard_lib.proto.schema_bp.TensorRepresentationRowPartitionDType attribute)
INT64 (merlin_standard_lib.proto.schema_bp.TensorRepresentationRowPartitionDType attribute)
int_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
int_value (merlin_standard_lib.proto.schema_bp.SequenceValueConstraints attribute)
(merlin_standard_lib.proto.schema_bp.TensorRepresentationDefaultValue attribute)
IntDomain (class in merlin_standard_lib.proto.schema_bp)
integer_format (merlin_standard_lib.proto.schema_bp.TimeDomain attribute)
(merlin_standard_lib.proto.schema_bp.TimeOfDayDomain attribute)
is_categorical (merlin_standard_lib.proto.schema_bp.IntDomain attribute)
is_embedding (merlin_standard_lib.proto.schema_bp.FloatDomain attribute)
is_gpu_dataloader_available() (in module transformers4rec.utils.dependencies)
is_merlin_dataloader_available() (in module transformers4rec.utils.dependencies)
is_pyarrow_available() (in module transformers4rec.utils.dependencies)
is_sorted (merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
item_embedding_table() (transformers4rec.torch.EmbeddingFeatures property)
(transformers4rec.torch.features.embedding.EmbeddingFeatures property)
(transformers4rec.torch.features.sequence.TabularSequenceFeatures property)
(transformers4rec.torch.TabularSequenceFeatures property)
item_id() (transformers4rec.torch.features.sequence.TabularSequenceFeatures property)
(transformers4rec.torch.TabularSequenceFeatures property)
item_id_column_name() (merlin_standard_lib.Schema property)
(merlin_standard_lib.schema.schema.Schema property)
(transformers4rec.torch.Schema property)
item_ids() (transformers4rec.torch.EmbeddingFeatures method)
(transformers4rec.torch.features.embedding.EmbeddingFeatures method)
J
jensen_shannon_divergence (merlin_standard_lib.proto.schema_bp.FeatureComparator attribute)
JensenShannonDivergence (class in merlin_standard_lib.proto.schema_bp)
L
LabelSmoothCrossEntropyLoss() (in module transformers4rec.torch)
LambdaModule (class in transformers4rec.torch.utils.torch_utils)
learning_rate_num_cosine_cycles_by_epoch (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
lifecycle_stage (merlin_standard_lib.proto.schema_bp.Feature attribute)
(merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
(merlin_standard_lib.proto.schema_bp.WeightedFeature attribute)
LifecycleStage (class in merlin_standard_lib.proto.schema_bp)
list_files() (in module transformers4rec.torch.utils.examples_utils)
load() (transformers4rec.torch.Model class method)
load_model_trainer_states_from_checkpoint() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
location_constraint_regex (merlin_standard_lib.proto.schema_bp.NaturalLanguageDomain attribute)
log() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
log_predictions (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
log_predictions_callback() (transformers4rec.torch.Trainer property)
(transformers4rec.torch.trainer.Trainer property)
LogUniformSampler (class in transformers4rec.torch.model.prediction_task)
LongformerConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
LossMixin (class in transformers4rec.torch.utils.torch_utils)
M
map_column_schemas() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
MappingTransformerMasking (class in transformers4rec.torch.utils.torch_utils)
MappingTransformerMasking.CausalLanguageModeling (class in transformers4rec.torch.utils.torch_utils)
MappingTransformerMasking.MaskedLanguageModeling (class in transformers4rec.torch.utils.torch_utils)
MappingTransformerMasking.PermutationLanguageModeling (class in transformers4rec.torch.utils.torch_utils)
MappingTransformerMasking.ReplacementLanguageModeling (class in transformers4rec.torch.utils.torch_utils)
MaskedLanguageModeling (class in transformers4rec.torch.masking)
masking() (transformers4rec.torch.features.sequence.TabularSequenceFeatures property)
(transformers4rec.torch.TabularSequenceFeatures property)
MaskingInfo (class in transformers4rec.torch.masking)
MaskSequence (class in transformers4rec.torch.masking)
max (merlin_standard_lib.proto.schema_bp.FloatDomain attribute)
(merlin_standard_lib.proto.schema_bp.IntDomain attribute)
(merlin_standard_lib.proto.schema_bp.UniqueConstraints attribute)
(merlin_standard_lib.proto.schema_bp.ValueCount attribute)
max_examples_count (merlin_standard_lib.proto.schema_bp.DatasetConstraints attribute)
max_fraction_of_sequences (merlin_standard_lib.proto.schema_bp.SequenceValueConstraints attribute)
max_fraction_threshold (merlin_standard_lib.proto.schema_bp.NumericValueComparator attribute)
max_image_byte_size (merlin_standard_lib.proto.schema_bp.ImageDomain attribute)
max_norm (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
max_per_sequence (merlin_standard_lib.proto.schema_bp.SequenceValueConstraints attribute)
max_sequence_length (merlin_standard_lib.proto.schema_bp.SequenceLengthConstraints attribute)
(transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.FeatureConfig attribute)
(transformers4rec.torch.features.embedding.FeatureConfig attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
MeanReciprocalRankAt (class in transformers4rec.torch.ranking_metric)
MegatronBertConfig (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
merge() (transformers4rec.torch.TabularModule method)
merge_values() (transformers4rec.torch.MergeTabular property)
MergeTabular (class in transformers4rec.torch)
merlin_standard_lib
module
merlin_standard_lib.proto
module
merlin_standard_lib.proto.schema_bp
module
merlin_standard_lib.schema
module
merlin_standard_lib.schema.schema
module
merlin_standard_lib.schema.tag
module
merlin_standard_lib.utils
module
merlin_standard_lib.utils.embedding_utils
module
MerlinDataLoader (class in transformers4rec.torch.utils.data_utils)
metadata() (merlin_standard_lib.proto.schema_bp.Annotation property)
metric_name() (transformers4rec.torch.PredictionTask method)
MetricsMixin (class in transformers4rec.torch.utils.torch_utils)
mid_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
MIDDomain (class in merlin_standard_lib.proto.schema_bp)
min (merlin_standard_lib.proto.schema_bp.FloatDomain attribute)
(merlin_standard_lib.proto.schema_bp.IntDomain attribute)
(merlin_standard_lib.proto.schema_bp.UniqueConstraints attribute)
(merlin_standard_lib.proto.schema_bp.ValueCount attribute)
min_avg_token_length (merlin_standard_lib.proto.schema_bp.FeatureCoverageConstraints attribute)
min_count (merlin_standard_lib.proto.schema_bp.FeaturePresence attribute)
min_coverage (merlin_standard_lib.proto.schema_bp.FeatureCoverageConstraints attribute)
min_domain_mass (merlin_standard_lib.proto.schema_bp.DistributionConstraints attribute)
min_examples_count (merlin_standard_lib.proto.schema_bp.DatasetConstraints attribute)
min_fraction (merlin_standard_lib.proto.schema_bp.FeaturePresence attribute)
min_fraction_of_sequences (merlin_standard_lib.proto.schema_bp.SequenceValueConstraints attribute)
min_fraction_threshold (merlin_standard_lib.proto.schema_bp.NumericValueComparator attribute)
min_per_sequence (merlin_standard_lib.proto.schema_bp.SequenceValueConstraints attribute)
min_sequence_length (merlin_standard_lib.proto.schema_bp.SequenceLengthConstraints attribute)
minimum_supported_image_fraction (merlin_standard_lib.proto.schema_bp.ImageDomain attribute)
MLPBlock (class in transformers4rec.torch)
(class in transformers4rec.torch.block.mlp)
mode (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
Model (class in transformers4rec.torch)
module
merlin_standard_lib
merlin_standard_lib.proto
merlin_standard_lib.proto.schema_bp
merlin_standard_lib.schema
merlin_standard_lib.schema.schema
merlin_standard_lib.schema.tag
merlin_standard_lib.utils
merlin_standard_lib.utils.embedding_utils
transformers4rec
transformers4rec.config
transformers4rec.config.schema
transformers4rec.config.trainer
transformers4rec.config.transformer
transformers4rec.torch
transformers4rec.torch.block
transformers4rec.torch.block.base
transformers4rec.torch.block.mlp
transformers4rec.torch.block.transformer
transformers4rec.torch.features
transformers4rec.torch.features.base
transformers4rec.torch.features.continuous
transformers4rec.torch.features.embedding
transformers4rec.torch.features.sequence
transformers4rec.torch.features.tabular
transformers4rec.torch.features.text
transformers4rec.torch.masking
transformers4rec.torch.model
transformers4rec.torch.model.prediction_task
transformers4rec.torch.ranking_metric
transformers4rec.torch.tabular
transformers4rec.torch.tabular.aggregation
transformers4rec.torch.tabular.transformations
transformers4rec.torch.trainer
transformers4rec.torch.typing
transformers4rec.torch.utils
transformers4rec.torch.utils.data_utils
transformers4rec.torch.utils.examples_utils
transformers4rec.torch.utils.schema_utils
transformers4rec.torch.utils.torch_utils
transformers4rec.types
transformers4rec.utils
transformers4rec.utils.dependencies
MPNetConfig (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
N
name (merlin_standard_lib.proto.schema_bp.BoolDomain attribute)
(merlin_standard_lib.proto.schema_bp.Feature attribute)
(merlin_standard_lib.proto.schema_bp.FixedShapeDim attribute)
(merlin_standard_lib.proto.schema_bp.FloatDomain attribute)
(merlin_standard_lib.proto.schema_bp.IntDomain attribute)
(merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
(merlin_standard_lib.proto.schema_bp.SparseFeatureIndexFeature attribute)
(merlin_standard_lib.proto.schema_bp.SparseFeatureValueFeature attribute)
(merlin_standard_lib.proto.schema_bp.StringDomain attribute)
(merlin_standard_lib.proto.schema_bp.WeightedFeature attribute)
(transformers4rec.torch.FeatureConfig attribute)
(transformers4rec.torch.features.embedding.FeatureConfig attribute)
(transformers4rec.torch.features.embedding.TableConfig attribute)
(transformers4rec.torch.TableConfig attribute)
natural_language_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
NaturalLanguageDomain (class in merlin_standard_lib.proto.schema_bp)
NDCGAt (class in transformers4rec.torch.ranking_metric)
nested_concat() (in module transformers4rec.torch.utils.torch_utils)
nested_detach() (in module transformers4rec.torch.utils.torch_utils)
nested_numpify() (in module transformers4rec.torch.utils.torch_utils)
nested_truncate() (in module transformers4rec.torch.utils.torch_utils)
NextItemPredictionPrepareBlock (class in transformers4rec.torch.model.prediction_task)
NextItemPredictionTask (class in transformers4rec.torch)
(class in transformers4rec.torch.model.prediction_task)
norm_type (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
not_in_environment (merlin_standard_lib.proto.schema_bp.Feature attribute)
num_embeddings (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
num_examples() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
num_examples_drift_comparator (merlin_standard_lib.proto.schema_bp.DatasetConstraints attribute)
num_examples_version_comparator (merlin_standard_lib.proto.schema_bp.DatasetConstraints attribute)
num_workers (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
NumericValueComparator (class in merlin_standard_lib.proto.schema_bp)
numpy_pad_and_concatenate() (in module transformers4rec.torch.utils.torch_utils)
O
on_epoch_end() (transformers4rec.torch.trainer.IncrementalLoggingCallback method)
on_train_begin() (transformers4rec.torch.trainer.IncrementalLoggingCallback method)
on_train_end() (transformers4rec.torch.trainer.IncrementalLoggingCallback method)
one_hot_1d() (in module transformers4rec.torch.utils.torch_utils)
oov_string_tokens (merlin_standard_lib.proto.schema_bp.FeatureCoverageConstraints attribute)
output_dir (transformers4rec.config.trainer.T4RecTrainingArgumentsTF attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
output_schema() (transformers4rec.torch.Model property)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader property)
output_size() (transformers4rec.torch.TabularBlock method)
(transformers4rec.torch.utils.torch_utils.OutputSizeMixin method)
OutputSizeMixin (class in transformers4rec.torch.utils.torch_utils)
P
PACKED_64_NANOS (merlin_standard_lib.proto.schema_bp.TimeOfDayDomainIntegerTimeOfDayFormat attribute)
pad_seq_column_if_needed() (transformers4rec.torch.utils.data_utils.ParquetDataset method)
padding_idx (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
ParquetDataset (class in transformers4rec.torch.utils.data_utils)
parse() (transformers4rec.torch.TabularAggregation class method)
(transformers4rec.torch.TabularTransformation class method)
(transformers4rec.torch.utils.data_utils.T4RecDataLoader class method)
parse_combiner() (transformers4rec.torch.features.embedding.PretrainedEmbeddingFeatures method)
(transformers4rec.torch.PretrainedEmbeddingFeatures method)
partition (merlin_standard_lib.proto.schema_bp.TensorRepresentationRaggedTensor attribute)
Path (class in merlin_standard_lib.proto.schema_bp)
PermutationLanguageModeling (class in transformers4rec.torch.masking)
pin_memory (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
pin_memory_device (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
place_model_on_device() (transformers4rec.config.trainer.T4RecTrainingArguments property)
(transformers4rec.torch.T4RecTrainingArguments property)
PLANNED (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
pop_labels() (transformers4rec.torch.Head method)
post() (transformers4rec.torch.TabularModule property)
post_forward() (transformers4rec.torch.TabularModule method)
pre() (transformers4rec.torch.TabularModule property)
pre_forward() (transformers4rec.torch.TabularModule method)
PrecisionAt (class in transformers4rec.torch.ranking_metric)
predict_all() (transformers4rec.torch.masking.MaskSequence method)
predict_top_k (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
prediction_step() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
prediction_tasks() (transformers4rec.torch.Model property)
PredictionTask (class in transformers4rec.torch)
prefetch_factor (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
prepare_module (transformers4rec.torch.block.transformer.TransformerBlock attribute)
presence (merlin_standard_lib.proto.schema_bp.Feature attribute)
(merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
PRETRAINED_EMBEDDING_MODULE_CLASS (transformers4rec.torch.features.tabular.TabularFeatures attribute)
(transformers4rec.torch.TabularFeatures attribute)
pretrained_module() (transformers4rec.torch.features.tabular.TabularFeatures property)
(transformers4rec.torch.TabularFeatures property)
PretrainedEmbeddingFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
PretrainedEmbeddingsInitializer (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
process_metrics() (in module transformers4rec.torch.trainer)
PRODUCTION (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
project_continuous_features() (transformers4rec.torch.features.sequence.TabularSequenceFeatures method)
(transformers4rec.torch.features.tabular.TabularFeatures method)
(transformers4rec.torch.TabularFeatures method)
(transformers4rec.torch.TabularSequenceFeatures method)
properties() (merlin_standard_lib.ColumnSchema property)
(merlin_standard_lib.schema.schema.ColumnSchema property)
PyarrowDataLoader (class in transformers4rec.torch.utils.data_utils)
R
ragged_tensor (merlin_standard_lib.proto.schema_bp.TensorRepresentation attribute)
random_data_from_schema() (in module transformers4rec.torch.utils.schema_utils)
RankingMetric (class in transformers4rec.torch.ranking_metric)
RecallAt (class in transformers4rec.torch.ranking_metric)
ReformerConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
RegressionPrepareBlock (class in transformers4rec.torch.model.prediction_task)
RegressionTask (class in transformers4rec.torch)
(class in transformers4rec.torch.model.prediction_task)
remove_by_name() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
remove_by_tag() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
remove_by_type() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
remove_pad_3d() (transformers4rec.torch.model.prediction_task.NextItemPredictionTask method)
(transformers4rec.torch.NextItemPredictionTask method)
ReplacementLanguageModeling (class in transformers4rec.torch.masking)
required (merlin_standard_lib.proto.schema_bp.FeaturePresenceWithinGroup attribute)
REQUIRES_SCHEMA (transformers4rec.config.schema.SchemaMixin attribute)
(transformers4rec.torch.ElementwiseSumItemMulti attribute)
(transformers4rec.torch.tabular.aggregation.ElementwiseSumItemMulti attribute)
requires_schema() (in module transformers4rec.config.schema)
(in module transformers4rec.torch)
(in module transformers4rec.torch.utils.torch_utils)
reset_lr_scheduler() (transformers4rec.torch.Trainer method)
(transformers4rec.torch.trainer.Trainer method)
reset_metrics() (transformers4rec.torch.Head method)
(transformers4rec.torch.Model method)
(transformers4rec.torch.PredictionTask method)
(transformers4rec.torch.utils.torch_utils.MetricsMixin method)
right_shift_block() (in module transformers4rec.torch)
(in module transformers4rec.torch.block.base)
RobertaConfig (class in transformers4rec.config.transformer)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
RoFormerConfig (transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
row_length (merlin_standard_lib.proto.schema_bp.TensorRepresentationRaggedTensorPartition attribute)
row_partition_dtype (merlin_standard_lib.proto.schema_bp.TensorRepresentationRaggedTensor attribute)
S
sample() (transformers4rec.torch.model.prediction_task.LogUniformSampler method)
sample_from_softmax() (transformers4rec.torch.masking.ReplacementLanguageModeling method)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking.ReplacementLanguageModeling method)
sampler (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
save() (transformers4rec.torch.Model method)
scale_grad_by_freq (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
Schema (class in merlin_standard_lib)
(class in merlin_standard_lib.schema.schema)
(class in transformers4rec.torch)
schema (transformers4rec.torch.masking.MaskingInfo attribute)
schema() (transformers4rec.config.schema.SchemaMixin property)
SchemaMixin (class in transformers4rec.config.schema)
select_by_name() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
select_by_tag() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
select_by_type() (merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
sequence_length_constraints (merlin_standard_lib.proto.schema_bp.NaturalLanguageDomain attribute)
SequenceEmbeddingFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.sequence)
SequenceLengthConstraints (class in merlin_standard_lib.proto.schema_bp)
SequenceValueConstraints (class in merlin_standard_lib.proto.schema_bp)
SequentialBlock (class in transformers4rec.torch)
(class in transformers4rec.torch.block.base)
SequentialTabularTransformations (class in transformers4rec.torch)
set_dataset() (transformers4rec.torch.utils.data_utils.MerlinDataLoader method)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader method)
(transformers4rec.torch.utils.data_utils.T4RecDataLoader method)
set_masking() (transformers4rec.torch.features.sequence.TabularSequenceFeatures method)
(transformers4rec.torch.TabularSequenceFeatures method)
set_metrics() (transformers4rec.torch.PredictionTask method)
set_schema() (transformers4rec.config.schema.SchemaMixin method)
shape (merlin_standard_lib.proto.schema_bp.Feature attribute)
(merlin_standard_lib.proto.schema_bp.TensorRepresentationDenseTensor attribute)
shuffle_buffer_size (transformers4rec.config.trainer.T4RecTrainingArguments attribute)
(transformers4rec.torch.T4RecTrainingArguments attribute)
ShuffleDataset (class in transformers4rec.torch.utils.data_utils)
size (merlin_standard_lib.proto.schema_bp.FixedShapeDim attribute)
skew_comparator (merlin_standard_lib.proto.schema_bp.Feature attribute)
SOFT_EMBEDDING_MODULE_CLASS (transformers4rec.torch.features.tabular.TabularFeatures attribute)
(transformers4rec.torch.TabularFeatures attribute)
SoftEmbedding (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
SoftEmbeddingFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
sparse (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
sparse_feature (merlin_standard_lib.proto.schema_bp.StructDomain attribute)
sparse_tensor (merlin_standard_lib.proto.schema_bp.TensorRepresentation attribute)
SparseFeature (class in merlin_standard_lib.proto.schema_bp)
SparseFeatureIndexFeature (class in merlin_standard_lib.proto.schema_bp)
SparseFeatureValueFeature (class in merlin_standard_lib.proto.schema_bp)
StackFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.aggregation)
step (merlin_standard_lib.proto.schema_bp.Path attribute)
StochasticSwapNoise (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.transformations)
string_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
string_format (merlin_standard_lib.proto.schema_bp.TimeDomain attribute)
(merlin_standard_lib.proto.schema_bp.TimeOfDayDomain attribute)
string_value (merlin_standard_lib.proto.schema_bp.SequenceValueConstraints attribute)
StringDomain (class in merlin_standard_lib.proto.schema_bp)
STRUCT (merlin_standard_lib.proto.schema_bp.FeatureType attribute)
struct_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
StructDomain (class in merlin_standard_lib.proto.schema_bp)
T
T4RecConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
T4RecDataLoader (class in transformers4rec.torch.utils.data_utils)
T4RecTrainingArguments (class in transformers4rec.config.trainer)
(class in transformers4rec.torch)
T4RecTrainingArgumentsTF (class in transformers4rec.config.trainer)
table (transformers4rec.torch.FeatureConfig attribute)
(transformers4rec.torch.features.embedding.FeatureConfig attribute)
table_to_embedding_module() (transformers4rec.torch.EmbeddingFeatures method)
(transformers4rec.torch.features.embedding.EmbeddingFeatures method)
(transformers4rec.torch.features.embedding.SoftEmbeddingFeatures method)
(transformers4rec.torch.features.sequence.SequenceEmbeddingFeatures method)
(transformers4rec.torch.SequenceEmbeddingFeatures method)
(transformers4rec.torch.SoftEmbeddingFeatures method)
TableConfig (class in transformers4rec.torch)
(class in transformers4rec.torch.features.embedding)
TabularAggregation (class in transformers4rec.torch)
TabularBlock (class in transformers4rec.torch)
TabularDropout (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.transformations)
TabularFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.tabular)
TabularLayerNorm (class in transformers4rec.torch)
(class in transformers4rec.torch.tabular.transformations)
TabularModule (class in transformers4rec.torch)
TabularSequenceFeatures (class in transformers4rec.torch)
(class in transformers4rec.torch.features.sequence)
TabularTransformation (class in transformers4rec.torch)
Tag (in module merlin_standard_lib.schema.tag)
tag (merlin_standard_lib.proto.schema_bp.Annotation attribute)
tags() (merlin_standard_lib.ColumnSchema property)
(merlin_standard_lib.schema.schema.ColumnSchema property)
targets (transformers4rec.torch.masking.MaskingInfo attribute)
task_blocks() (transformers4rec.torch.Head property)
task_name() (transformers4rec.torch.PredictionTask property)
tensor_representation (merlin_standard_lib.proto.schema_bp.TensorRepresentationGroup attribute)
TensorRepresentation (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationDefaultValue (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationDenseTensor (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationGroup (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationRaggedTensor (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationRaggedTensorPartition (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationRowPartitionDType (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationSparseTensor (class in merlin_standard_lib.proto.schema_bp)
TensorRepresentationVarLenSparseTensor (class in merlin_standard_lib.proto.schema_bp)
threshold (merlin_standard_lib.proto.schema_bp.InfinityNorm attribute)
(merlin_standard_lib.proto.schema_bp.JensenShannonDivergence attribute)
time_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
time_of_day_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
TimeDomain (class in merlin_standard_lib.proto.schema_bp)
TimeDomainIntegerTimeFormat (class in merlin_standard_lib.proto.schema_bp)
TimeOfDayDomain (class in merlin_standard_lib.proto.schema_bp)
TimeOfDayDomainIntegerTimeOfDayFormat (class in merlin_standard_lib.proto.schema_bp)
timeout (transformers4rec.torch.utils.data_utils.DLDataLoader attribute)
(transformers4rec.torch.utils.data_utils.MerlinDataLoader attribute)
(transformers4rec.torch.utils.data_utils.PyarrowDataLoader attribute)
to_core_schema() (in module transformers4rec.torch.utils.data_utils)
to_head() (transformers4rec.torch.PredictionTask method)
to_huggingface_torch_model() (transformers4rec.config.transformer.T4RecConfig method)
(transformers4rec.torch.T4RecConfig method)
to_lightning() (transformers4rec.torch.Model method)
to_model() (transformers4rec.torch.block.base.BlockBase method)
(transformers4rec.torch.BlockBase method)
(transformers4rec.torch.Head method)
(transformers4rec.torch.PredictionTask method)
to_module() (transformers4rec.torch.block.base.BuildableBlock method)
(transformers4rec.torch.TabularBlock method)
to_proto_text() (merlin_standard_lib.ColumnSchema method)
(merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.ColumnSchema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
to_torch_model() (transformers4rec.config.transformer.T4RecConfig method)
(transformers4rec.torch.T4RecConfig method)
token_constraints (merlin_standard_lib.proto.schema_bp.NaturalLanguageDomain attribute)
torch_pad_and_concatenate() (in module transformers4rec.torch.utils.torch_utils)
Trainer (class in transformers4rec.torch)
(class in transformers4rec.torch.trainer)
training (transformers4rec.torch.BinaryClassificationTask attribute)
(transformers4rec.torch.block.transformer.GPT2Prepare attribute)
(transformers4rec.torch.block.transformer.TransformerBlock attribute)
(transformers4rec.torch.block.transformer.TransformerPrepare attribute)
(transformers4rec.torch.features.embedding.PretrainedEmbeddingsInitializer attribute)
(transformers4rec.torch.features.embedding.SoftEmbedding attribute)
(transformers4rec.torch.Head attribute)
(transformers4rec.torch.Model attribute)
(transformers4rec.torch.model.prediction_task.BinaryClassificationTask attribute)
(transformers4rec.torch.model.prediction_task.LogUniformSampler attribute)
(transformers4rec.torch.model.prediction_task.NextItemPredictionTask attribute)
(transformers4rec.torch.model.prediction_task.RegressionTask attribute)
(transformers4rec.torch.NextItemPredictionTask attribute)
(transformers4rec.torch.PredictionTask attribute)
(transformers4rec.torch.PretrainedEmbeddingsInitializer attribute)
(transformers4rec.torch.RegressionTask attribute)
(transformers4rec.torch.SoftEmbedding attribute)
(transformers4rec.torch.tabular.aggregation.ElementwiseSumItemMulti attribute)
(transformers4rec.torch.TabularModule attribute)
(transformers4rec.torch.utils.torch_utils.LambdaModule attribute)
tranform_label_to_onehot() (in module transformers4rec.torch.utils.torch_utils)
transformer (transformers4rec.torch.block.transformer.TransformerBlock attribute)
transformer_arguments() (transformers4rec.torch.masking.MaskSequence property)
transformer_optional_arguments() (transformers4rec.torch.masking.MaskSequence method)
transformer_required_arguments() (transformers4rec.torch.masking.MaskSequence method)
(transformers4rec.torch.masking.PermutationLanguageModeling method)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking.PermutationLanguageModeling method)
TRANSFORMER_TO_PREPARE (transformers4rec.torch.block.transformer.TransformerBlock attribute)
(transformers4rec.torch.TransformerBlock attribute)
TransformerBlock (class in transformers4rec.torch)
(class in transformers4rec.torch.block.transformer)
TransformerPrepare (class in transformers4rec.torch.block.transformer)
transformers4rec
module
transformers4rec.config
module
transformers4rec.config.schema
module
transformers4rec.config.trainer
module
transformers4rec.config.transformer
module
transformers4rec.torch
module
transformers4rec.torch.block
module
transformers4rec.torch.block.base
module
transformers4rec.torch.block.mlp
module
transformers4rec.torch.block.transformer
module
transformers4rec.torch.features
module
transformers4rec.torch.features.base
module
transformers4rec.torch.features.continuous
module
transformers4rec.torch.features.embedding
module
transformers4rec.torch.features.sequence
module
transformers4rec.torch.features.tabular
module
transformers4rec.torch.features.text
module
transformers4rec.torch.masking
module
transformers4rec.torch.model
module
transformers4rec.torch.model.prediction_task
module
transformers4rec.torch.ranking_metric
module
transformers4rec.torch.tabular
module
transformers4rec.torch.tabular.aggregation
module
transformers4rec.torch.tabular.transformations
module
transformers4rec.torch.trainer
module
transformers4rec.torch.typing
module
transformers4rec.torch.utils
module
transformers4rec.torch.utils.data_utils
module
transformers4rec.torch.utils.examples_utils
module
transformers4rec.torch.utils.schema_utils
module
transformers4rec.torch.utils.torch_utils
module
transformers4rec.types
module
transformers4rec.utils
module
transformers4rec.utils.dependencies
module
transformers_config_cls() (transformers4rec.config.transformer.T4RecConfig property)
(transformers4rec.torch.T4RecConfig property)
TransfoXLConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
true_value (merlin_standard_lib.proto.schema_bp.BoolDomain attribute)
type (merlin_standard_lib.proto.schema_bp.Feature attribute)
(merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
TYPE_UNKNOWN (merlin_standard_lib.proto.schema_bp.FeatureType attribute)
U
uint_value (merlin_standard_lib.proto.schema_bp.TensorRepresentationDefaultValue attribute)
uniform_row_length (merlin_standard_lib.proto.schema_bp.TensorRepresentationRaggedTensorPartition attribute)
unique_constraints (merlin_standard_lib.proto.schema_bp.Feature attribute)
UniqueConstraints (class in merlin_standard_lib.proto.schema_bp)
UNIX_DAYS (merlin_standard_lib.proto.schema_bp.TimeDomainIntegerTimeFormat attribute)
UNIX_MICROSECONDS (merlin_standard_lib.proto.schema_bp.TimeDomainIntegerTimeFormat attribute)
UNIX_MILLISECONDS (merlin_standard_lib.proto.schema_bp.TimeDomainIntegerTimeFormat attribute)
UNIX_NANOSECONDS (merlin_standard_lib.proto.schema_bp.TimeDomainIntegerTimeFormat attribute)
UNIX_SECONDS (merlin_standard_lib.proto.schema_bp.TimeDomainIntegerTimeFormat attribute)
UNKNOWN_STAGE (merlin_standard_lib.proto.schema_bp.LifecycleStage attribute)
UNSPECIFIED (merlin_standard_lib.proto.schema_bp.TensorRepresentationRowPartitionDType attribute)
update() (transformers4rec.torch.ranking_metric.RankingMetric method)
url_domain (merlin_standard_lib.proto.schema_bp.Feature attribute)
URLDomain (class in merlin_standard_lib.proto.schema_bp)
V
value (merlin_standard_lib.proto.schema_bp.StringDomain attribute)
value_column_name (merlin_standard_lib.proto.schema_bp.TensorRepresentationSparseTensor attribute)
value_count (merlin_standard_lib.proto.schema_bp.Feature attribute)
(merlin_standard_lib.proto.schema_bp.ValueCountList attribute)
value_counts (merlin_standard_lib.proto.schema_bp.Feature attribute)
value_feature (merlin_standard_lib.proto.schema_bp.SparseFeature attribute)
ValueCount (class in merlin_standard_lib.proto.schema_bp)
ValueCountList (class in merlin_standard_lib.proto.schema_bp)
varlen_sparse_tensor (merlin_standard_lib.proto.schema_bp.TensorRepresentation attribute)
visualize_response() (in module transformers4rec.torch.utils.examples_utils)
vocabulary (merlin_standard_lib.proto.schema_bp.NaturalLanguageDomain attribute)
vocabulary_size (transformers4rec.torch.features.embedding.TableConfig attribute)
(transformers4rec.torch.TableConfig attribute)
W
weight (transformers4rec.torch.features.embedding.EmbeddingBagWrapper attribute)
weight_feature (merlin_standard_lib.proto.schema_bp.WeightedFeature attribute)
WeightedFeature (class in merlin_standard_lib.proto.schema_bp)
wipe_memory() (in module transformers4rec.torch.utils.examples_utils)
with_name() (merlin_standard_lib.ColumnSchema method)
(merlin_standard_lib.schema.schema.ColumnSchema method)
with_properties() (merlin_standard_lib.ColumnSchema method)
(merlin_standard_lib.schema.schema.ColumnSchema method)
with_tags() (merlin_standard_lib.ColumnSchema method)
(merlin_standard_lib.schema.schema.ColumnSchema method)
with_tags_based_on_properties() (merlin_standard_lib.ColumnSchema method)
(merlin_standard_lib.Schema method)
(merlin_standard_lib.schema.schema.ColumnSchema method)
(merlin_standard_lib.schema.schema.Schema method)
(transformers4rec.torch.Schema method)
X
XLNetConfig (class in transformers4rec.config.transformer)
(class in transformers4rec.torch)
(transformers4rec.torch.utils.torch_utils.MappingTransformerMasking attribute)
Other Versions
v: stable
Tags
v23.02.00
v23.04.00
v23.05.00
v23.06.00
v23.08.00
v23.12.00
Branches
main
stable