Index

_

__init__() (recis.features.feature.Feature method) (recis.features.FeatureEngine method) (recis.features.op.Bucketize method) (recis.features.op.FeatureCross method) (recis.features.op.Hash method) (recis.features.op.IDMultiHash method) (recis.features.op.Mod method) (recis.features.op.SelectField method) (recis.features.op.SelectFields method) (recis.features.op.SequenceTruncate method) (recis.fg.feature_generator.FG method) (recis.fg.fg_parser.FGParser method) (recis.fg.mc_parser.MCParser method) (recis.framework.checkpoint_manager.Saver method) (recis.framework.exporter.Exporter method) (recis.framework.model_bank.ModelBankParser method) (recis.framework.trainer.Trainer method) (recis.hooks.filter_hook.HashTableFilterHook method) (recis.hooks.metric_report_hook.MetricReportHook method) (recis.hooks.ml_tracker_hook.MLTrackerHook method) (recis.hooks.profiler_hook.ProfilerHook method) (recis.hooks.trace_to_odps_hook.TraceToOdpsHook method) (recis.io.dataset_base.DatasetBase method) (recis.io.OrcDataset method) (recis.metrics.auroc.AUROC method) (recis.metrics.gauc.Gauc method) (recis.nn.DynamicEmbedding method) (recis.nn.EmbeddingEngine method) (recis.nn.modules.hashtable.HashTable method) (recis.optim.adamw_tf.AdamWTF method) (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method) (recis.serialize.checkpoint_reader.CheckpointReader method) (recis.serialize.loader.Loader method) (recis.serialize.saver.Saver method) (recis.utils.logger.Logger method)

_aucs (recis.metrics.gauc.Gauc attribute) _counts (recis.metrics.gauc.Gauc attribute) _cpu (recis.metrics.gauc.Gauc attribute) _ids (recis.fg.feature_generator.FG attribute) _labels (recis.fg.feature_generator.FG attribute) _word_size (recis.metrics.gauc.Gauc attribute)

A

accelerator (recis.framework.trainer.Trainer attribute) accept_grad() (recis.nn.modules.hashtable.HashTable method) AdamWTF (class in recis.optim.adamw_tf) add_hook() (recis.framework.trainer.Trainer method) add_hooks() (recis.framework.trainer.Trainer method) add_id() (recis.fg.feature_generator.FG method) add_io_features() (recis.fg.feature_generator.FG method) add_label() (recis.fg.feature_generator.FG method) add_metric() (in module recis.framework.metrics) add_op() (recis.features.feature.Feature method) add_path() (recis.io.OrcDataset method)

add_paths() (recis.io.OrcDataset method) add_to_ml_tracker() (in module recis.hooks.ml_tracker_hook) add_to_trace() (in module recis.hooks.trace_to_odps_hook) admit_hook (recis.nn.EmbeddingOption attribute) AdmitHook (class in recis.nn.hashtable_hook) after_data() (recis.hooks.hook.Hook method) after_epoch() (recis.hooks.hook.Hook method) after_step() (recis.hooks.hook.Hook method) after_window() (recis.hooks.hook.Hook method) already_hashed (recis.fg.fg_parser.FGParser attribute) args (recis.framework.trainer.Trainer attribute) AUROC (class in recis.metrics.auroc)

B

batch_size (recis.io.dataset_base.DatasetBase attribute) before_epoch() (recis.hooks.hook.Hook method) before_step() (recis.hooks.hook.Hook method) before_window() (recis.hooks.hook.Hook method) block_shapes (recis.fg.feature_generator.FG property) block_size (recis.nn.EmbeddingOption attribute)

blocks (recis.fg.mc_parser.MCParser attribute) boundary (recis.features.op.Bucketize attribute) Bucketize (class in recis.features.op) bucketize() (in module recis.nn.functional.array_ops) bucketize_mod() (in module recis.nn.functional.array_ops) build_fg() (in module recis.fg.feature_generator)

C

CheckpointReader (class in recis.serialize.checkpoint_reader) children (recis.nn.EmbeddingOption attribute) children_info() (recis.nn.modules.hashtable.HashTable method) ckpt_dir (recis.framework.exporter.Exporter attribute) ckpt_load_arg (recis.framework.trainer.TrainingArguments attribute) ckpt_save_arg (recis.framework.trainer.TrainingArguments attribute)

clear_grad() (recis.nn.modules.hashtable.HashTable method) coalesced (recis.nn.EmbeddingOption attribute) combiner (recis.nn.EmbeddingOption attribute) combiner_kwargs (recis.nn.EmbeddingOption attribute) compute() (recis.metrics.auroc.AUROC method) ConstantInitializer (class in recis.nn.initializers) copy_data_to_device() (in module recis.utils.data_utils)

D

data_to_cuda (recis.framework.trainer.Trainer attribute) dataset (recis.framework.exporter.Exporter attribute) DatasetBase (class in recis.io.dataset_base) dense_lr_scheduler (recis.framework.trainer.Trainer attribute) dense_model (recis.framework.exporter.Exporter attribute) dense_optimizer (recis.framework.exporter.Exporter attribute) (recis.framework.trainer.Trainer attribute) devel_mode (recis.fg.fg_parser.FGParser attribute)

device (recis.io.dataset_base.DatasetBase attribute) (recis.nn.EmbeddingOption attribute) dist_sync_on_step (recis.metrics.auroc.AUROC attribute) drop_remainder (recis.io.dataset_base.DatasetBase attribute) dtype (recis.io.dataset_base.DatasetBase attribute) (recis.nn.EmbeddingOption attribute) dump_io_state() (recis.io.dataset_base.DatasetBase method) DynamicEmbedding (class in recis.nn)

E

emb_conf_ (recis.fg.fg_parser.FGParser attribute) emb_configs (recis.fg.fg_parser.FGParser property) emb_default_class (recis.fg.feature_generator.FG attribute) emb_default_device (recis.fg.feature_generator.FG attribute) emb_default_type (recis.fg.feature_generator.FG attribute) embedding_dim (recis.nn.EmbeddingOption attribute) embedding_initializer (recis.fg.feature_generator.FG attribute) EmbeddingEngine (class in recis.nn) EmbeddingOption (class in recis.nn) embeddings() (recis.nn.modules.hashtable.HashTable method)

end() (recis.hooks.hook.Hook method) error() (recis.utils.logger.Logger method) eval_dataset (recis.framework.trainer.Trainer attribute) eval_steps (recis.framework.trainer.TrainingArguments attribute) evaluate() (recis.framework.trainer.Trainer method) export() (recis.framework.exporter.Exporter method) export_dir (recis.framework.exporter.Exporter attribute) export_model_name (recis.framework.exporter.Exporter attribute) export_outputs (recis.framework.exporter.Exporter attribute) Exporter (class in recis.framework.exporter)

F

farmhash() (in module recis.nn.functional.hash_ops) fea_blocks (recis.fg.mc_parser.MCParser attribute) Feature (class in recis.features.feature) feature_blocks (recis.fg.feature_generator.FG property) (recis.fg.fg_parser.FGParser property) (recis.fg.mc_parser.MCParser property) feature_shapes (recis.fg.feature_generator.FG property) FeatureCross (class in recis.features.op) FeatureEngine (class in recis.features) FG (class in recis.fg.feature_generator) fg_conf (recis.fg.fg_parser.FGParser attribute) (recis.framework.exporter.Exporter attribute) fg_parser (recis.fg.feature_generator.FG attribute) FGParser (class in recis.fg.fg_parser) filter_hook (recis.nn.EmbeddingOption attribute) filter_out_sparse_param() (in module recis.nn.modules.hashtable) filter_sparse_opt (recis.framework.exporter.Exporter attribute) FilterHook (class in recis.nn.hashtable_hook) fixedlen_feature() (recis.io.dataset_base.DatasetBase method) fn (recis.metrics.auroc.AUROC attribute) forward() (recis.features.feature.Feature method) (recis.features.FeatureEngine method) (recis.features.op.FeatureCross method) (recis.features.op.Hash method) (recis.features.op.IDMultiHash method) (recis.features.op.Mod method) (recis.features.op.SelectField method) (recis.features.op.SelectFields method) (recis.features.op.SequenceTruncate method) (recis.metrics.auroc.AUROC method) (recis.metrics.gauc.Gauc method) (recis.nn.DynamicEmbedding method) (recis.nn.EmbeddingEngine method) (recis.nn.modules.hashtable.HashTable method)

fp (recis.metrics.auroc.AUROC attribute) fused_bucketize_gpu() (in module recis.nn.functional.fused_ops) fused_ids_encode_gpu() (in module recis.nn.functional.fused_ops) fused_multi_hash() (in module recis.nn.functional.fused_ops), [1] fused_ragged_cutoff_2D() (in module recis.nn.functional.ragged_ops) fused_ragged_cutoff_3D() (in module recis.nn.functional.ragged_ops) fused_uint64_mod_gpu() (in module recis.nn.functional.fused_ops) fx_tool (recis.framework.exporter.Exporter attribute)

G

Gauc (class in recis.metrics.gauc) gen_slice() (in module recis.nn.modules.hashtable) get_block_seq_len() (recis.fg.feature_generator.FG method) get_emb_confs() (recis.fg.feature_generator.FG method) get_feature_confs() (recis.fg.feature_generator.FG method)

get_seq_len() (recis.fg.fg_parser.FGParser method) get_shape() (recis.fg.feature_generator.FG method) grad() (recis.nn.modules.hashtable.HashTable method) grad_reduce_by (recis.fg.feature_generator.FG attribute) (recis.nn.EmbeddingOption attribute) gradient_accumulation_steps (recis.framework.trainer.TrainingArguments attribute)

H

has_fea() (recis.fg.mc_parser.MCParser method) has_seq_fea() (recis.fg.mc_parser.MCParser method) Hash (class in recis.features.op) hash_buckets (recis.io.OrcDataset attribute) hash_features (recis.io.OrcDataset attribute) hash_in_io (recis.fg.fg_parser.FGParser attribute)

hash_type (recis.features.op.Hash attribute) hash_types (recis.io.OrcDataset attribute) HashTable (class in recis.nn.modules.hashtable) HashTableFilterHook (class in recis.hooks.filter_hook) Hook (class in recis.hooks.hook) hooks (recis.framework.trainer.Trainer attribute)

I

IDMultiHash (class in recis.features.op) ids() (recis.nn.modules.hashtable.HashTable method) ids_encode() (in module recis.nn.functional.embedding_ops) ids_partition() (in module recis.nn.functional.embedding_ops) info() (recis.utils.logger.Logger method) init_kwargs (recis.fg.feature_generator.FG attribute)

initializer (recis.nn.EmbeddingOption attribute) input_list (recis.features.op.SelectFields attribute) insert() (recis.nn.modules.hashtable.HashTable method) io_conf_ (recis.fg.fg_parser.FGParser attribute) io_configs (recis.fg.fg_parser.FGParser property) is_compressed (recis.io.dataset_base.DatasetBase attribute) is_seq_block() (recis.fg.feature_generator.FG method)

L

labels (recis.fg.feature_generator.FG property) load() (recis.serialize.loader.Loader method) load_io_state() (recis.io.dataset_base.DatasetBase method) load_update_epochs (recis.framework.trainer.TrainingArguments attribute) load_update_steps (recis.framework.trainer.TrainingArguments attribute)

load_update_windows (recis.framework.trainer.TrainingArguments attribute) Loader (class in recis.serialize.loader) log_steps (recis.framework.trainer.TrainingArguments attribute) Logger (class in recis.utils.logger) logger (recis.hooks.profiler_hook.ProfilerHook attribute) LoggerHook (class in recis.hooks.logger_hook)

M

max_partition_num (recis.nn.EmbeddingOption attribute) max_to_keep (recis.framework.trainer.TrainingArguments attribute) mc_conf (recis.fg.mc_parser.MCParser attribute) (recis.framework.exporter.Exporter attribute) mc_parser (recis.fg.fg_parser.FGParser attribute) MCParser (class in recis.fg.mc_parser) merge_offsets() (in module recis.nn.functional.embedding_ops) MetricReportHook (class in recis.hooks.metric_report_hook) mixed_precision (recis.framework.trainer.TrainingArguments attribute) MLTrackerHook (class in recis.hooks.ml_tracker_hook) Mod (class in recis.features.op) mod (recis.features.op.Mod attribute)

model (recis.framework.exporter.Exporter attribute) (recis.framework.trainer.Trainer attribute) model_bank (recis.framework.trainer.TrainingArguments attribute) ModelBankParser (class in recis.framework.model_bank) module recis.nn.functional.embedding_ops recis.nn.functional.fused_ops recis.nn.functional.hash_ops recis.nn.functional.ragged_ops multi_hash() (in module recis.nn.functional.array_ops) multihash_conf_ (recis.fg.fg_parser.FGParser attribute) murmurhash() (in module recis.nn.functional.hash_ops)

N

NormalInitializer (class in recis.nn.initializers)

num_thresholds (recis.metrics.auroc.AUROC attribute)

O

OrcDataset (class in recis.io) out_off_data() (recis.hooks.hook.Hook method)

output_dir (recis.framework.trainer.TrainingArguments attribute) (recis.hooks.profiler_hook.ProfilerHook attribute)

P

pack_threads_num (recis.io.dataset_base.DatasetBase attribute) params_not_save (recis.framework.trainer.TrainingArguments attribute) parsed_conf_ (recis.fg.fg_parser.FGParser attribute)

pg (recis.nn.EmbeddingOption attribute) prefetch (recis.io.dataset_base.DatasetBase attribute) prof (recis.hooks.profiler_hook.ProfilerHook attribute) ProfilerHook (class in recis.hooks.profiler_hook)

Q

queue (recis.hooks.trace_to_odps_hook.TraceToOdpsHook attribute)

R

ragged_embedding_segment_reduce() (in module recis.nn.functional.embedding_ops) ragged_format (recis.io.dataset_base.DatasetBase attribute) rank (recis.framework.exporter.Exporter attribute) read_tensor() (recis.serialize.checkpoint_reader.CheckpointReader method) read_threads_num (recis.io.dataset_base.DatasetBase attribute) reader (recis.serialize.checkpoint_reader.CheckpointReader attribute) recis.nn.functional.embedding_ops module recis.nn.functional.fused_ops module

recis.nn.functional.hash_ops module recis.nn.functional.ragged_ops module register_for_checkpointing() (recis.framework.checkpoint_manager.Saver method) register_io_state() (recis.framework.checkpoint_manager.Saver method) reset() (recis.io.dataset_base.DatasetBase method) (recis.metrics.auroc.AUROC method) (recis.metrics.gauc.Gauc method)

S

sample_ids (recis.fg.feature_generator.FG property) save() (recis.framework.checkpoint_manager.Saver method) (recis.serialize.saver.Saver method) save_concurrency_per_rank (recis.framework.trainer.TrainingArguments attribute) save_end (recis.framework.trainer.TrainingArguments attribute) save_every_n_epochs (recis.framework.trainer.TrainingArguments attribute) save_every_n_windows (recis.framework.trainer.TrainingArguments attribute) save_filter_fn (recis.framework.trainer.TrainingArguments attribute) save_interval (recis.io.dataset_base.DatasetBase attribute) save_steps (recis.framework.trainer.TrainingArguments attribute) Saver (class in recis.framework.checkpoint_manager) (class in recis.serialize.saver) saver_option (recis.framework.trainer.TrainingArguments attribute) SelectField (class in recis.features.op) SelectFields (class in recis.features.op) seq_block_names (recis.fg.feature_generator.FG property) (recis.fg.fg_parser.FGParser property) (recis.fg.mc_parser.MCParser property) seq_blocks (recis.fg.mc_parser.MCParser attribute)

SequenceTruncate (class in recis.features.op) set_grad_accum_steps() (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method) shape_manager (recis.fg.feature_generator.FG attribute) shard_num (recis.framework.exporter.Exporter attribute) shared_name (recis.nn.EmbeddingOption attribute) slot_group() (recis.nn.modules.hashtable.HashTable method) sparse_embedding_segment_reduce() (in module recis.nn.functional.embedding_ops) sparse_model (recis.framework.exporter.Exporter attribute) sparse_optimizer (recis.framework.trainer.Trainer attribute) SparseAdam (class in recis.optim.sparse_adam) SparseAdamW (class in recis.optim.sparse_adamw) SparseAdamWTF (class in recis.optim.sparse_adamw_tf) start() (recis.hooks.hook.Hook method) step() (recis.optim.adamw_tf.AdamWTF method) (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method)

T

tensor_dtype() (recis.serialize.checkpoint_reader.CheckpointReader method) tensor_names() (recis.serialize.checkpoint_reader.CheckpointReader method) tensor_shape() (recis.serialize.checkpoint_reader.CheckpointReader method) thresholds (recis.metrics.auroc.AUROC attribute) tn (recis.metrics.auroc.AUROC attribute) tp (recis.metrics.auroc.AUROC attribute) TraceToOdpsHook (class in recis.hooks.trace_to_odps_hook) tracker (recis.hooks.ml_tracker_hook.MLTrackerHook attribute) train() (recis.framework.trainer.Trainer method)

train_and_evaluate() (recis.framework.trainer.Trainer method) train_dataset (recis.framework.trainer.Trainer attribute) train_epoch (recis.framework.trainer.TrainingArguments attribute) train_steps (recis.framework.trainer.TrainingArguments attribute) trainable (recis.nn.EmbeddingOption attribute) Trainer (class in recis.framework.trainer) TrainingArguments (class in recis.framework.trainer) transform_fn (recis.io.dataset_base.DatasetBase attribute) TruncNormalInitializer (class in recis.nn.initializers)

U

UniformInitializer (class in recis.nn.initializers) update() (recis.metrics.auroc.AUROC method)

use_coalesce (recis.fg.feature_generator.FG attribute) use_weight (recis.nn.EmbeddingOption attribute) uses_columns (recis.fg.mc_parser.MCParser attribute)

V

varlen_feature() (recis.io.dataset_base.DatasetBase method)

W

warning() (recis.utils.logger.Logger method) window_mode() (recis.hooks.hook.Hook method) worker_idx (recis.io.dataset_base.DatasetBase attribute)

worker_num (recis.io.dataset_base.DatasetBase attribute) worker_slice_batch_num (recis.io.dataset_base.DatasetBase attribute) writer_num (recis.hooks.trace_to_odps_hook.TraceToOdpsHook attribute) writers (recis.hooks.trace_to_odps_hook.TraceToOdpsHook attribute)

Z

zero_grad() (recis.optim.adamw_tf.AdamWTF method) (recis.optim.sparse_adam.SparseAdam method) (recis.optim.sparse_adamw.SparseAdamW method) (recis.optim.sparse_adamw_tf.SparseAdamWTF method)