Index

A

ACR (class in ml4ir.applications.ranking.model.metrics.metrics_impl) add_fn() (ml4ir.base.features.feature_layer.FeatureLayerMap method) add_fns() (ml4ir.base.features.feature_layer.FeatureLayerMap method) add_temperature_layer() (ml4ir.base.model.relevance_model.RelevanceModel method)

all_features (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) architecture_op() (ml4ir.base.model.scoring.scoring_model.RelevanceScorer method) (ml4ir.base.model.scoring.scoring_model.ScorerBase method)

B

balance_classes() (ml4ir.base.data.relevance_dataset.RelevanceDataset method)

build() (ml4ir.base.features.feature_fns.categorical.VocabLookup method) bytes_sequence_to_encoding_bilstm() (in module ml4ir.base.features.feature_fns.sequence)

C

calibrate() (ml4ir.base.model.relevance_model.RelevanceModel method) call() (ml4ir.base.features.feature_fns.categorical.CategoricalDropout method) (ml4ir.base.features.feature_fns.categorical.VocabLookup method) categorical_embedding_to_encoding_bilstm() (in module ml4ir.base.features.feature_fns.categorical) categorical_embedding_with_hash_buckets() (in module ml4ir.base.features.feature_fns.categorical) categorical_embedding_with_indices() (in module ml4ir.base.features.feature_fns.categorical) categorical_embedding_with_vocabulary_file() (in module ml4ir.base.features.feature_fns.categorical) categorical_embedding_with_vocabulary_file_and_dropout() (in module ml4ir.base.features.feature_fns.categorical) categorical_indicator_with_vocabulary_file() (in module ml4ir.base.features.feature_fns.categorical) categorical_indices_from_vocabulary_file() (in module ml4ir.base.features.feature_fns.categorical) CategoricalAccuracy (class in ml4ir.applications.classification.model.metrics.metrics_impl) CategoricalCrossEntropy (class in ml4ir.applications.classification.model.losses.categorical_cross_entropy) CategoricalDropout (class in ml4ir.base.features.feature_fns.categorical)

ClassificationPipeline (class in ml4ir.applications.classification.pipeline) clear_dir() (ml4ir.base.io.file_io.FileIO method) clear_dir_contents() (ml4ir.base.io.local_io.LocalIO method) context_features (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) copy_from_hdfs() (ml4ir.base.io.spark_io.SparkIO method) copy_to_hdfs() (ml4ir.base.io.spark_io.SparkIO method) create_dataset() (ml4ir.base.data.relevance_dataset.RelevanceDataset method) create_dummy_protobuf() (ml4ir.base.features.feature_config.ExampleFeatureConfig method) (ml4ir.base.features.feature_config.FeatureConfig method) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) create_pipeline_for_kfold() (ml4ir.applications.classification.pipeline.ClassificationPipeline method) (ml4ir.applications.ranking.pipeline.RankingPipeline method) (ml4ir.base.pipeline.RelevancePipeline method)

D

default_value (ml4ir.base.features.feature_fns.categorical.VocabLookup attribute) define_feature_layer() (in module ml4ir.base.features.feature_layer)

define_inputs() (ml4ir.base.features.feature_config.FeatureConfig method) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) define_scheduler_as_callback() (ml4ir.base.model.relevance_model.RelevanceModel method)

E

evaluate() (ml4ir.applications.ranking.model.ranking_model.RankingModel method) (ml4ir.base.model.relevance_model.RelevanceModel method) ExampleFeatureConfig (class in ml4ir.base.features.feature_config) extract_features() (ml4ir.base.features.feature_config.FeatureConfig method) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method)

extract_features_from_proto() (ml4ir.base.data.tfrecord_reader.TFRecordExampleParser method) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) (ml4ir.base.data.tfrecord_reader.TFRecordSequenceExampleParser method)

F

feature_exists() (ml4ir.base.features.feature_config.FeatureConfig method) feature_layer_op() (ml4ir.base.model.scoring.interaction_model.InteractionModel method) (ml4ir.base.model.scoring.interaction_model.UnivariateInteractionModel method) feature_name (ml4ir.base.features.feature_fns.categorical.VocabLookup attribute) FeatureConfig (class in ml4ir.base.features.feature_config) FeatureLayerMap (class in ml4ir.base.features.feature_layer) features (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) features_dict (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute)

features_to_log (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) FileIO (class in ml4ir.base.io.file_io) final_activation_op() (ml4ir.base.model.scoring.scoring_model.RelevanceScorer method) (ml4ir.base.model.scoring.scoring_model.ScorerBase method) finish() (ml4ir.base.pipeline.RelevancePipeline method) fit() (ml4ir.base.model.relevance_model.RelevanceModel method) from_model_config_file() (ml4ir.base.model.scoring.scoring_model.ScorerBase class method) from_relevance_scorer() (ml4ir.base.model.relevance_model.RelevanceModel class method) from_univariate_interaction_model() (ml4ir.base.model.relevance_model.RelevanceModel class method)

G

generate_and_add_mask() (ml4ir.base.data.tfrecord_reader.TFRecordExampleParser method) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) (ml4ir.base.data.tfrecord_reader.TFRecordSequenceExampleParser method) generate_mask() (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) get_all_features() (ml4ir.base.features.feature_config.FeatureConfig method) get_bilstm_encoding() (in module ml4ir.base.features.feature_fns.sequence) get_config() (ml4ir.base.features.feature_fns.categorical.CategoricalDropout method) (ml4ir.base.features.feature_fns.categorical.VocabLookup method) get_context_features() (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) get_default_tensor() (ml4ir.base.data.tfrecord_reader.TFRecordExampleParser method) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) (ml4ir.base.data.tfrecord_reader.TFRecordSequenceExampleParser method) get_default_value() (ml4ir.base.features.feature_config.FeatureConfig method) get_dtype() (ml4ir.base.features.feature_config.FeatureConfig method) get_feature() (ml4ir.base.data.tfrecord_reader.TFRecordExampleParser method) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) (ml4ir.base.data.tfrecord_reader.TFRecordSequenceExampleParser method) (ml4ir.base.features.feature_config.FeatureConfig method) get_feature_by_node_name() (ml4ir.base.features.feature_config.FeatureConfig method) get_features_spec() (ml4ir.base.data.tfrecord_reader.TFRecordExampleParser method) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) (ml4ir.base.data.tfrecord_reader.TFRecordSequenceExampleParser method) get_features_to_log() (ml4ir.base.features.feature_config.FeatureConfig method) get_files_in_directory() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) get_final_activation_op() (ml4ir.applications.classification.model.losses.categorical_cross_entropy.CategoricalCrossEntropy method) (ml4ir.applications.ranking.model.losses.pointwise_losses.SigmoidCrossEntropy method) (ml4ir.base.model.losses.loss_base.RelevanceLossBase method) get_fn() (ml4ir.base.features.feature_layer.FeatureLayerMap method) get_fns() (ml4ir.base.features.feature_layer.FeatureLayerMap method) get_group_metrics_keys() (ml4ir.base.features.feature_config.FeatureConfig method)

get_hyperparameter_dict() (ml4ir.base.features.feature_config.FeatureConfig method) get_instance() (ml4ir.base.features.feature_config.FeatureConfig static method) get_kfold_relevance_dataset() (ml4ir.applications.classification.pipeline.ClassificationPipeline method) (ml4ir.base.pipeline.RelevancePipeline method) get_label() (ml4ir.base.features.feature_config.FeatureConfig method) get_loss_fn() (ml4ir.applications.classification.model.losses.categorical_cross_entropy.CategoricalCrossEntropy method) (ml4ir.applications.ranking.model.losses.listwise_losses.RankOneListNet method) (ml4ir.applications.ranking.model.losses.pointwise_losses.SigmoidCrossEntropy method) (ml4ir.base.model.losses.loss_base.RelevanceLossBase method) get_mask() (ml4ir.base.features.feature_config.FeatureConfig method) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) get_metadata_features() (ml4ir.base.features.feature_config.FeatureConfig method) get_one_hot_label_vectorizer() (in module ml4ir.base.features.preprocessing) get_parse_fn() (in module ml4ir.base.data.tfrecord_reader) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) get_path_from_str() (ml4ir.base.io.spark_io.SparkIO method) get_query_key() (ml4ir.base.features.feature_config.FeatureConfig method) get_rank() (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) get_relevance_dataset() (ml4ir.applications.classification.pipeline.ClassificationPipeline method) (ml4ir.base.pipeline.RelevancePipeline method) get_relevance_model() (ml4ir.applications.classification.pipeline.ClassificationPipeline method) (ml4ir.applications.ranking.pipeline.RankingPipeline method) (ml4ir.base.pipeline.RelevancePipeline method) get_secondary_labels() (ml4ir.base.features.feature_config.FeatureConfig method) get_sequence_features() (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) get_train_features() (ml4ir.base.features.feature_config.FeatureConfig method) get_vocabulary_info() (in module ml4ir.base.features.feature_fns.categorical) global_1d_pooling() (in module ml4ir.base.features.feature_fns.sequence) group_metrics_keys (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute)

I

initialize_features() (ml4ir.base.features.feature_config.FeatureConfig method) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method)

InteractionModel (class in ml4ir.base.model.scoring.interaction_model) is_compiled (ml4ir.base.model.relevance_model.RelevanceModel attribute)

K

kfold_analysis() (ml4ir.applications.ranking.pipeline.RankingPipeline method)

L

label (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) load() (ml4ir.base.model.relevance_model.RelevanceModel method) load_numpy_array() (ml4ir.base.io.local_io.LocalIO method) load_weights() (ml4ir.base.model.relevance_model.RelevanceModel method) LocalIO (class in ml4ir.base.io.local_io)

log() (ml4ir.base.io.file_io.FileIO method) log_initialization() (ml4ir.base.features.feature_config.FeatureConfig method) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig method) logger (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) lookup_table (ml4ir.base.features.feature_fns.categorical.VocabLookup attribute)

M

make_directory() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) mask (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) metadata_features (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) ml4ir.base.data.csv_reader (module)

ml4ir.base.data.tfrecord_reader (module) ml4ir.base.data.tfrecord_writer (module) ml4ir.base.features.feature_fns.categorical (module) ml4ir.base.features.feature_fns.sequence (module) ml4ir.base.features.feature_fns.tf_native (module) ml4ir.base.features.feature_layer (module) ml4ir.base.features.preprocessing (module) MRR (class in ml4ir.applications.ranking.model.metrics.metrics_impl)

N

natural_log (in module ml4ir.base.features.preprocessing)

num_oov_buckets (ml4ir.base.features.feature_fns.categorical.VocabLookup attribute)

P

pad_feature() (ml4ir.base.data.tfrecord_reader.TFRecordExampleParser method) (ml4ir.base.data.tfrecord_reader.TFRecordParser method) (ml4ir.base.data.tfrecord_reader.TFRecordSequenceExampleParser method) path_exists() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method)

pop_fn() (ml4ir.base.features.feature_layer.FeatureLayerMap method) post_training_step() (ml4ir.base.pipeline.RelevancePipeline method) pre_processing_step() (ml4ir.base.pipeline.RelevancePipeline method) predict() (ml4ir.applications.ranking.model.ranking_model.RankingModel method) (ml4ir.base.model.relevance_model.RelevanceModel method) preprocess_feature() (ml4ir.base.data.tfrecord_reader.TFRecordParser method) preprocess_text (in module ml4ir.base.features.preprocessing)

Q

query_key (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute)

R

rank (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) RankingModel (class in ml4ir.applications.ranking.model.ranking_model) RankingPipeline (class in ml4ir.applications.ranking.pipeline) RankOneListNet (class in ml4ir.applications.ranking.model.losses.listwise_losses) read() (in module ml4ir.base.data.csv_reader) (in module ml4ir.base.data.tfrecord_reader) read_df() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method) read_df_list() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method) read_json() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method) read_text_file() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.spark_io.SparkIO method) read_yaml() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method)

RelevanceDataset (class in ml4ir.base.data.relevance_dataset) RelevanceLossBase (class in ml4ir.base.model.losses.loss_base) RelevanceModel (class in ml4ir.base.model.relevance_model) RelevancePipeline (class in ml4ir.base.pipeline) RelevanceScorer (class in ml4ir.base.model.scoring.scoring_model) rm_dir() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method) rm_file() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) (ml4ir.base.io.spark_io.SparkIO method) run() (ml4ir.base.pipeline.RelevancePipeline method) run_kfold_analysis() (ml4ir.applications.classification.pipeline.ClassificationPipeline method) (ml4ir.applications.ranking.pipeline.RankingPipeline method) run_pipeline() (ml4ir.base.pipeline.RelevancePipeline method) run_ttest() (ml4ir.applications.ranking.model.ranking_model.RankingModel method) (ml4ir.base.model.relevance_model.RelevanceModel method)

S

save() (ml4ir.applications.ranking.model.ranking_model.RankingModel method) (ml4ir.base.model.relevance_model.RelevanceModel method) save_numpy_array() (ml4ir.base.io.local_io.LocalIO method) ScorerBase (class in ml4ir.base.model.scoring.scoring_model) secondary_labels (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) sequence_features (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute)

SequenceExampleFeatureConfig (class in ml4ir.base.features.feature_config) set_feature() (ml4ir.base.features.feature_config.FeatureConfig method) set_logger() (ml4ir.base.io.file_io.FileIO method) set_seeds() (ml4ir.base.pipeline.RelevancePipeline method) setup_logging() (ml4ir.base.pipeline.RelevancePipeline method) SigmoidCrossEntropy (class in ml4ir.applications.ranking.model.losses.pointwise_losses) SparkIO (class in ml4ir.base.io.spark_io) split_and_pad_string (in module ml4ir.base.features.preprocessing)

T

tf_native_op() (in module ml4ir.base.features.feature_fns.tf_native) TFRecordExampleParser (class in ml4ir.base.data.tfrecord_reader) TFRecordParser (class in ml4ir.base.data.tfrecord_reader) TFRecordSequenceExampleParser (class in ml4ir.base.data.tfrecord_reader) Top5CategoricalAccuracy (class in ml4ir.applications.classification.model.metrics.metrics_impl)

train_features (ml4ir.base.features.feature_config.ExampleFeatureConfig attribute) (ml4ir.base.features.feature_config.FeatureConfig attribute) (ml4ir.base.features.feature_config.SequenceExampleFeatureConfig attribute) train_val_test_split() (ml4ir.base.data.relevance_dataset.RelevanceDataset method) transform_features_op() (ml4ir.base.model.scoring.interaction_model.InteractionModel method) (ml4ir.base.model.scoring.interaction_model.UnivariateInteractionModel method)

U

UnivariateInteractionModel (class in ml4ir.base.model.scoring.interaction_model)

update_state() (ml4ir.applications.classification.model.metrics.metrics_impl.Top5CategoricalAccuracy method)

V

validate_args() (ml4ir.applications.ranking.pipeline.RankingPipeline method)

VocabLookup (class in ml4ir.base.features.feature_fns.categorical) vocabulary_list (ml4ir.base.features.feature_fns.categorical.VocabLookup attribute)

W

write_df() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method) write_from_df() (in module ml4ir.base.data.tfrecord_writer)

write_from_files() (in module ml4ir.base.data.tfrecord_writer) write_json() (ml4ir.base.io.file_io.FileIO method) (ml4ir.base.io.local_io.LocalIO method)