Index

_

__call__() (dicee.models.base_model.IdentityClass method) (dicee.models.ensemble.EnsembleKGE method) (dicee.models.IdentityClass method), [1], [2] __getitem__() (dicee.dataset_classes.AllvsAll method) (dicee.dataset_classes.BPE_NegativeSamplingDataset method) (dicee.dataset_classes.KvsAll method) (dicee.dataset_classes.KvsSampleDataset method) (dicee.dataset_classes.LiteralDataset method) (dicee.dataset_classes.MultiClassClassificationDataset method) (dicee.dataset_classes.MultiLabelDataset method) (dicee.dataset_classes.NegSampleDataset method) (dicee.dataset_classes.OnevsAllDataset method) (dicee.dataset_classes.OnevsSample method) (dicee.dataset_classes.TriplePredictionDataset method) __iter__() (dicee.config.Namespace method) (dicee.knowledge_graph.KG method) (dicee.models.ensemble.EnsembleKGE method) __len__() (dicee.dataset_classes.AllvsAll method) (dicee.dataset_classes.BPE_NegativeSamplingDataset method) (dicee.dataset_classes.KvsAll method) (dicee.dataset_classes.KvsSampleDataset method) (dicee.dataset_classes.LiteralDataset method) (dicee.dataset_classes.MultiClassClassificationDataset method) (dicee.dataset_classes.MultiLabelDataset method) (dicee.dataset_classes.NegSampleDataset method) (dicee.dataset_classes.OnevsAllDataset method) (dicee.dataset_classes.OnevsSample method) (dicee.dataset_classes.TriplePredictionDataset method) (dicee.knowledge_graph.KG method) (dicee.models.ensemble.EnsembleKGE method)

__setstate__() (dicee.models.ADOPT method) (dicee.models.adopt.ADOPT method) __str__() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) (dicee.models.ensemble.EnsembleKGE method) __version__ (in module dicee)

A

AbstractCallback (class in dicee.abstracts) AbstractPPECallback (class in dicee.abstracts) AbstractTrainer (class in dicee.abstracts) AccumulateEpochLossCallback (class in dicee.callbacks) achieve_answer() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) AConEx (class in dicee.models) (class in dicee.models.complex) AConvO (class in dicee.models) (class in dicee.models.octonion) AConvQ (class in dicee.models) (class in dicee.models.quaternion) adaptive_lr (dicee.config.Namespace attribute) adaptive_swa (dicee.config.Namespace attribute) add_new_entity_embeddings() (dicee.abstracts.BaseInteractiveKGE method) add_noise_rate (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) add_noisy_triples() (in module dicee.static_funcs) add_noisy_triples_into_training() (dicee.read_preprocess_save_load_kg.read_from_disk.ReadFromDisk method) (dicee.read_preprocess_save_load_kg.ReadFromDisk method) add_reciprocal (dicee.knowledge_graph.KG attribute) ADOPT (class in dicee.models) (class in dicee.models.adopt) adopt() (in module dicee.models.adopt) AllvsAll (class in dicee.dataset_classes) alphas (dicee.abstracts.AbstractPPECallback attribute) (dicee.weight_averaging.ASWA attribute) analyse() (in module dicee.analyse_experiments) answer_multi_hop_query() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) app (in module dicee.scripts.index_serve)

apply_coefficients() (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.DeCaL method) (dicee.models.Keci method) apply_reciprocal_or_noise() (in module dicee.read_preprocess_save_load_kg.util) apply_semantic_constraint (dicee.abstracts.BaseInteractiveKGE attribute) apply_unit_norm (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] args (dicee.DICE_Trainer attribute) (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] (dicee.models.base_model.BaseKGE attribute) (dicee.models.base_model.IdentityClass attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] (dicee.models.ensemble.EnsembleKGE attribute) (dicee.models.IdentityClass attribute), [1], [2] (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) ASWA (class in dicee.weight_averaging) aswa (dicee.analyse_experiments.Experiment attribute) attn (dicee.models.Block attribute) (dicee.models.transformers.Block attribute) attn_dropout (dicee.models.transformers.SelfAttention attribute) attributes (dicee.abstracts.AbstractTrainer attribute) auto_batch_finding (dicee.config.Namespace attribute)

B

backend (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) base_weights (dicee.weight_averaging.TWA attribute) BaseInteractiveKGE (class in dicee.abstracts) BaseInteractiveTrainKGE (class in dicee.abstracts) BaseKGE (class in dicee.models), [1], [2], [3], [4], [5], [6] (class in dicee.models.base_model) BaseKGELightning (class in dicee.models) (class in dicee.models.base_model) batch_kronecker_product() (dicee.callbacks.KronE static method) batch_size (dicee.analyse_experiments.Experiment attribute) (dicee.callbacks.PseudoLabellingCallback attribute) (dicee.config.Namespace attribute) (dicee.dataset_classes.CVDataModule attribute) batches_per_epoch (dicee.callbacks.LRScheduler attribute) beta (dicee.weight_averaging.TWA attribute) bias (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) (dicee.models.transformers.LayerNorm attribute) Block (class in dicee.models) (class in dicee.models.transformers) block_size (dicee.config.Namespace attribute) (dicee.dataset_classes.MultiClassClassificationDataset attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute)

blocks (dicee.models.CoKE attribute) (dicee.models.real.CoKE attribute) bn_conv1 (dicee.models.AConvQ attribute) (dicee.models.ConvQ attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) bn_conv2 (dicee.models.AConvQ attribute) (dicee.models.ConvQ attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) bn_conv2d (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) BPE_NegativeSamplingDataset (class in dicee.dataset_classes) build_chain_funcs() (dicee.models.FMult2 method) (dicee.models.function_space.FMult2 method) build_func() (dicee.models.FMult2 method) (dicee.models.function_space.FMult2 method) build_projection() (dicee.weight_averaging.TWA method) BytE (class in dicee.models.transformers) byte_pair_encoding (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6]

C

c_attn (dicee.models.transformers.SelfAttention attribute) c_fc (dicee.models.transformers.MLP attribute) c_proj (dicee.models.transformers.MLP attribute) (dicee.models.transformers.SelfAttention attribute) callbacks (dicee.abstracts.AbstractTrainer attribute) (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) causal (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) (dicee.models.transformers.SelfAttention attribute) chain_func() (dicee.models.FMult method) (dicee.models.function_space.FMult method) (dicee.models.function_space.GFMult method) (dicee.models.GFMult method) CKeci (class in dicee.models) (class in dicee.models.clifford) cl_pqr() (dicee.models.clifford.DeCaL method) (dicee.models.DeCaL method) cleanup() (dicee.Execute method) (dicee.executer.Execute method) clifford_multiplication() (dicee.models.clifford.Keci method) (dicee.models.Keci method) clip_lambda (dicee.models.ADOPT attribute) (dicee.models.adopt.ADOPT attribute) CoKE (class in dicee.models) (class in dicee.models.real) coke_dropout (dicee.models.CoKE attribute) (dicee.models.real.CoKE attribute) CoKEConfig (class in dicee.models) (class in dicee.models.real) collate_fn (dicee.dataset_classes.AllvsAll attribute) (dicee.dataset_classes.KvsAll attribute) (dicee.dataset_classes.KvsSampleDataset attribute) (dicee.dataset_classes.MultiClassClassificationDataset attribute) (dicee.dataset_classes.MultiLabelDataset attribute) (dicee.dataset_classes.OnevsAllDataset attribute) (dicee.dataset_classes.OnevsSample attribute), [1] collate_fn() (dicee.dataset_classes.BPE_NegativeSamplingDataset method) (dicee.dataset_classes.TriplePredictionDataset method) collection_name (dicee.scripts.index_serve.NeuralSearcher attribute) comp_func() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) ComplEx (class in dicee.models) (class in dicee.models.complex) compute_convergence() (in module dicee.callbacks) compute_func() (dicee.models.FMult method) (dicee.models.FMult2 method) (dicee.models.function_space.FMult method) (dicee.models.function_space.FMult2 method) (dicee.models.function_space.GFMult method) (dicee.models.GFMult method) compute_metrics_from_ranks() (in module dicee.evaluation) (in module dicee.evaluation.utils) compute_metrics_from_ranks_simple() (in module dicee.evaluation.utils) compute_mrr() (dicee.weight_averaging.ASWA static method) compute_sigma_pp() (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.DeCaL method) (dicee.models.Keci method) compute_sigma_pq() (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.DeCaL method) (dicee.models.Keci method) compute_sigma_pr() (dicee.models.clifford.DeCaL method) (dicee.models.DeCaL method) compute_sigma_qq() (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.DeCaL method) (dicee.models.Keci method) compute_sigma_qr() (dicee.models.clifford.DeCaL method) (dicee.models.DeCaL method)

compute_sigma_rr() (dicee.models.clifford.DeCaL method) (dicee.models.DeCaL method) compute_sigmas_multivect() (dicee.models.clifford.DeCaL method) (dicee.models.DeCaL method) compute_sigmas_single() (dicee.models.clifford.DeCaL method) (dicee.models.DeCaL method) ConEx (class in dicee.models) (class in dicee.models.complex) config (dicee.models.CoKE attribute) (dicee.models.real.CoKE attribute) (dicee.models.transformers.BytE attribute) (dicee.models.transformers.GPT attribute) configs (dicee.abstracts.BaseInteractiveKGE attribute) configure_optimizers() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) (dicee.models.transformers.GPT method) construct_batch_selected_cl_multivector() (dicee.models.clifford.Keci method) (dicee.models.Keci method) construct_cl_multivector() (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.DeCaL method) (dicee.models.Keci method) construct_dataset() (in module dicee.dataset_classes) construct_ensemble (dicee.abstracts.BaseInteractiveKGE attribute) construct_graph() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) construct_input_and_output() (dicee.abstracts.BaseInteractiveKGE method) construct_multi_coeff() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) continual_learning (dicee.config.Namespace attribute) continual_start() (dicee.DICE_Trainer method) (dicee.executer.ContinuousExecute method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) continual_training_setup_executor() (in module dicee.static_funcs) ContinuousExecute (class in dicee.executer) conv2d (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.AConvQ attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.ConvQ attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) ConvO (class in dicee.models) (class in dicee.models.octonion) ConvQ (class in dicee.models) (class in dicee.models.quaternion) count_triples() (in module dicee.read_preprocess_save_load_kg.util) create_and_store_kg() (dicee.Execute method) (dicee.executer.Execute method) create_constraints() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_preprocess_funcs) create_experiment_folder() (in module dicee.static_funcs) create_random_data() (dicee.callbacks.PseudoLabellingCallback method) create_recipriocal_triples() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) create_vector_database() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) crop_block_size() (dicee.models.transformers.GPT method) ctx (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) current_epoch (dicee.weight_averaging.EMA attribute) (dicee.weight_averaging.SWA attribute) (dicee.weight_averaging.SWAG attribute) (dicee.weight_averaging.TWA attribute) CVDataModule (class in dicee.dataset_classes) cycle_length (dicee.callbacks.LRScheduler attribute)

D

data_module (dicee.callbacks.PseudoLabellingCallback attribute) data_property_embeddings (dicee.models.literal.LiteralEmbeddings attribute) data_property_to_idx (dicee.dataset_classes.LiteralDataset attribute) dataset_dir (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute), [1] dataset_sanity_checking() (in module dicee.read_preprocess_save_load_kg.util) DeCaL (class in dicee.models) (class in dicee.models.clifford) decay (dicee.weight_averaging.EMA attribute) decide() (dicee.weight_averaging.ASWA method) default_eval_model (dicee.callbacks.PeriodicEvalCallback attribute) degree (dicee.models.function_space.LFMult attribute) (dicee.models.LFMult attribute) denormalize() (dicee.dataset_classes.LiteralDataset static method) describe() (dicee.knowledge_graph.KG method) description_of_input (dicee.knowledge_graph.KG attribute) deviations (dicee.weight_averaging.SWAG attribute) device (dicee.models.literal.LiteralEmbeddings property) DICE_Trainer (class in dicee) (class in dicee.trainer) (class in dicee.trainer.dice_trainer) dicee module dicee.__main__ module dicee.abstracts module dicee.analyse_experiments module dicee.callbacks module dicee.config module dicee.dataset_classes module dicee.eval_static_funcs module dicee.evaluation module dicee.evaluation.ensemble module dicee.evaluation.evaluator module dicee.evaluation.link_prediction module dicee.evaluation.literal_prediction module dicee.evaluation.utils module dicee.evaluator module dicee.executer module dicee.knowledge_graph module dicee.knowledge_graph_embeddings module dicee.models module dicee.models.adopt module dicee.models.base_model module dicee.models.clifford module dicee.models.complex module dicee.models.dualE module dicee.models.ensemble module dicee.models.function_space module dicee.models.literal module dicee.models.octonion module dicee.models.pykeen_models module

dicee.models.quaternion module dicee.models.real module dicee.models.static_funcs module dicee.models.transformers module dicee.query_generator module dicee.read_preprocess_save_load_kg module dicee.read_preprocess_save_load_kg.preprocess module dicee.read_preprocess_save_load_kg.read_from_disk module dicee.read_preprocess_save_load_kg.save_load_disk module dicee.read_preprocess_save_load_kg.util module dicee.sanity_checkers module dicee.scripts module dicee.scripts.index_serve module dicee.scripts.run module dicee.static_funcs module dicee.static_funcs_training module dicee.static_preprocess_funcs module dicee.trainer module dicee.trainer.dice_trainer module dicee.trainer.model_parallelism module dicee.trainer.torch_trainer module dicee.trainer.torch_trainer_ddp module dicee.weight_averaging module discrete_points (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult2 attribute) dist_func (dicee.models.Pyke attribute) (dicee.models.real.Pyke attribute) DistMult (class in dicee.models) (class in dicee.models.real) distributed (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] domain_constraints_per_rel (dicee.evaluation.Evaluator attribute) (dicee.evaluation.evaluator.Evaluator attribute) (dicee.Evaluator attribute) (dicee.evaluator.Evaluator attribute) download_file() (in module dicee.static_funcs) download_files_from_url() (in module dicee.static_funcs) download_pretrained_model() (in module dicee.static_funcs) dropout (dicee.models.CoKEConfig attribute), [1] (dicee.models.literal.LiteralEmbeddings attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) (dicee.models.transformers.MLP attribute) (dicee.models.transformers.SelfAttention attribute) DualE (class in dicee.models) (class in dicee.models.dualE) dummy_eval() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) dummy_id (dicee.knowledge_graph.KG attribute) during_training (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1]

E

ee_vocab (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] efficient_zero_grad() (in module dicee.evaluation.utils) (in module dicee.static_funcs_training) EMA (class in dicee.weight_averaging) ema (dicee.config.Namespace attribute) ema_c_epochs (dicee.weight_averaging.EMA attribute) ema_model (dicee.weight_averaging.EMA attribute) ema_start_epoch (dicee.weight_averaging.EMA attribute) ema_update() (dicee.weight_averaging.EMA static method) embedding_dim (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] (dicee.models.literal.LiteralEmbeddings attribute) embedding_dims (dicee.models.literal.LiteralEmbeddings attribute) enable_log (in module dicee.static_preprocess_funcs) enc (dicee.knowledge_graph.KG attribute) end() (dicee.Execute method) (dicee.executer.Execute method) EnsembleKGE (class in dicee.models.ensemble) ent2id (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) ent_in (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) ent_out (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) entities_str (dicee.knowledge_graph.KG property) entity_embeddings (dicee.models.AConvQ attribute) (dicee.models.clifford.DeCaL attribute) (dicee.models.ConvQ attribute) (dicee.models.DeCaL attribute) (dicee.models.DualE attribute) (dicee.models.dualE.DualE attribute) (dicee.models.FMult attribute) (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.FMult2 attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.function_space.LFMult attribute) (dicee.models.function_space.LFMult1 attribute) (dicee.models.GFMult attribute) (dicee.models.LFMult attribute) (dicee.models.LFMult1 attribute) (dicee.models.literal.LiteralEmbeddings attribute), [1] (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) entity_to_idx (dicee.dataset_classes.LiteralDataset attribute), [1] (dicee.knowledge_graph.KG attribute), [1] (dicee.scripts.index_serve.NeuralSearcher attribute) epoch_count (dicee.abstracts.AbstractPPECallback attribute) (dicee.weight_averaging.ASWA attribute) epoch_counter (dicee.callbacks.Eval attribute) (dicee.callbacks.KGESaveCallback attribute) (dicee.callbacks.PeriodicEvalCallback attribute) epoch_ratio (dicee.callbacks.Eval attribute) er_vocab (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] estimate_mfu() (dicee.models.transformers.GPT method) estimate_q() (in module dicee.callbacks) Eval (class in dicee.callbacks) eval() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) (dicee.models.ensemble.EnsembleKGE method) eval_at_epochs (dicee.config.Namespace attribute) eval_epochs (dicee.callbacks.PeriodicEvalCallback attribute) eval_every_n_epochs (dicee.config.Namespace attribute) eval_lp_performance() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) eval_model (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) eval_rank_of_head_and_tail_byte_pair_encoded_entity() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) eval_rank_of_head_and_tail_entity() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method)

eval_with_bpe_vs_all() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) eval_with_byte() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) eval_with_data() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) eval_with_vs_all() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) evaluate() (in module dicee.static_funcs) evaluate_bpe_lp() (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) (in module dicee.static_funcs_training) evaluate_ensemble_link_prediction_performance() (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.ensemble) evaluate_link_prediction_performance() (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) evaluate_link_prediction_performance_with_bpe() (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) evaluate_link_prediction_performance_with_bpe_reciprocals() (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) evaluate_link_prediction_performance_with_reciprocals() (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) evaluate_literal_prediction() (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.literal_prediction) evaluate_lp() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) (in module dicee.static_funcs_training) evaluate_lp_bpe_k_vs_all() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) (in module dicee.eval_static_funcs) (in module dicee.evaluation) (in module dicee.evaluation.link_prediction) evaluate_lp_k_vs_all() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) evaluate_lp_with_byte() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) Evaluator (class in dicee) (class in dicee.evaluation) (class in dicee.evaluation.evaluator) (class in dicee.evaluator) evaluator (dicee.DICE_Trainer attribute) (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) every_x_epoch (dicee.callbacks.KGESaveCallback attribute) example_input_array (dicee.models.ensemble.EnsembleKGE property) Execute (class in dicee) (class in dicee.executer) exists() (dicee.knowledge_graph.KG method) Experiment (class in dicee.analyse_experiments) experiment_dir (dicee.callbacks.LRScheduler attribute) (dicee.callbacks.PeriodicEvalCallback attribute) explicit (dicee.models.QMult attribute) (dicee.models.quaternion.QMult attribute) exponential_function() (in module dicee.static_funcs) extract_input_outputs() (dicee.trainer.torch_trainer_ddp.NodeTrainer method) (in module dicee.trainer.model_parallelism) extract_input_outputs_set_device() (dicee.trainer.torch_trainer.TorchTrainer method)

F

f (dicee.callbacks.KronE attribute) fc (dicee.models.literal.LiteralEmbeddings attribute) fc1 (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.AConvQ attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.ConvQ attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) fc_num_input (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.AConvQ attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.ConvQ attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) fc_out (dicee.models.literal.LiteralEmbeddings attribute) feature_map_dropout (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.AConvQ attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.ConvQ attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) feature_map_dropout_rate (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] fetch_worker() (in module dicee.read_preprocess_save_load_kg.util) fill_query() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) find_good_batch_size() (in module dicee.trainer.model_parallelism) find_missing_triples() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) fit() (dicee.trainer.model_parallelism.TensorParallel method) (dicee.trainer.torch_trainer.TorchTrainer method) (dicee.trainer.torch_trainer_ddp.TorchDDPTrainer method) flash (dicee.models.transformers.SelfAttention attribute) FMult (class in dicee.models) (class in dicee.models.function_space) FMult2 (class in dicee.models) (class in dicee.models.function_space) form_of_labelling (dicee.DICE_Trainer attribute) (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) forward() (dicee.models.base_model.BaseKGE method) (dicee.models.base_model.IdentityClass static method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] (dicee.models.Block method) (dicee.models.IdentityClass static method), [1], [2] (dicee.models.literal.LiteralEmbeddings method) (dicee.models.transformers.Block method) (dicee.models.transformers.BytE method) (dicee.models.transformers.GPT method) (dicee.models.transformers.LayerNorm method) (dicee.models.transformers.MLP method) (dicee.models.transformers.SelfAttention method) forward_backward_update() (dicee.trainer.torch_trainer.TorchTrainer method) forward_backward_update_loss() (in module dicee.trainer.model_parallelism) forward_byte_pair_encoded_k_vs_all() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] forward_byte_pair_encoded_triple() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] forward_k_vs_all() (dicee.models.AConEx method) (dicee.models.AConvO method) (dicee.models.AConvQ method) (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.CoKE method) (dicee.models.ComplEx method) (dicee.models.complex.AConEx method) (dicee.models.complex.ComplEx method) (dicee.models.complex.ConEx method) (dicee.models.ConEx method) (dicee.models.ConvO method) (dicee.models.ConvQ method) (dicee.models.DeCaL method) (dicee.models.DistMult method) (dicee.models.DualE method) (dicee.models.dualE.DualE method) (dicee.models.Keci method) (dicee.models.octonion.AConvO method) (dicee.models.octonion.ConvO method) (dicee.models.octonion.OMult method) (dicee.models.OMult method) (dicee.models.pykeen_models.PykeenKGE method) (dicee.models.PykeenKGE method) (dicee.models.QMult method) (dicee.models.quaternion.AConvQ method) (dicee.models.quaternion.ConvQ method) (dicee.models.quaternion.QMult method) (dicee.models.real.CoKE method) (dicee.models.real.DistMult method) (dicee.models.real.Shallom method) (dicee.models.real.TransE method) (dicee.models.Shallom method) (dicee.models.TransE method)

forward_k_vs_sample() (dicee.models.AConEx method) (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] (dicee.models.clifford.Keci method) (dicee.models.CoKE method) (dicee.models.ComplEx method) (dicee.models.complex.AConEx method) (dicee.models.complex.ComplEx method) (dicee.models.complex.ConEx method) (dicee.models.ConEx method) (dicee.models.DistMult method) (dicee.models.Keci method) (dicee.models.pykeen_models.PykeenKGE method) (dicee.models.PykeenKGE method) (dicee.models.QMult method) (dicee.models.quaternion.QMult method) (dicee.models.real.CoKE method) (dicee.models.real.DistMult method) forward_k_vs_with_explicit() (dicee.models.clifford.Keci method) (dicee.models.Keci method) forward_triples() (dicee.models.AConEx method) (dicee.models.AConvO method) (dicee.models.AConvQ method) (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] (dicee.models.clifford.DeCaL method) (dicee.models.clifford.Keci method) (dicee.models.complex.AConEx method) (dicee.models.complex.ConEx method) (dicee.models.ConEx method) (dicee.models.ConvO method) (dicee.models.ConvQ method) (dicee.models.DeCaL method) (dicee.models.DualE method) (dicee.models.dualE.DualE method) (dicee.models.FMult method) (dicee.models.FMult2 method) (dicee.models.function_space.FMult method) (dicee.models.function_space.FMult2 method) (dicee.models.function_space.GFMult method) (dicee.models.function_space.LFMult method) (dicee.models.function_space.LFMult1 method) (dicee.models.GFMult method) (dicee.models.Keci method) (dicee.models.LFMult method) (dicee.models.LFMult1 method) (dicee.models.octonion.AConvO method) (dicee.models.octonion.ConvO method) (dicee.models.Pyke method) (dicee.models.pykeen_models.PykeenKGE method) (dicee.models.PykeenKGE method) (dicee.models.quaternion.AConvQ method) (dicee.models.quaternion.ConvQ method) (dicee.models.real.Pyke method) (dicee.models.real.Shallom method) (dicee.models.Shallom method) freeze_entity_embeddings (dicee.models.literal.LiteralEmbeddings attribute), [1] frequency (dicee.callbacks.Perturb attribute) from_pretrained() (dicee.models.transformers.GPT class method) from_pretrained_model_write_embeddings_into_csv() (in module dicee.static_funcs) full_storage_path (dicee.analyse_experiments.Experiment attribute) func_triple_to_bpe_representation (dicee.evaluation.Evaluator attribute) (dicee.evaluation.evaluator.Evaluator attribute) (dicee.Evaluator attribute) (dicee.evaluator.Evaluator attribute) func_triple_to_bpe_representation() (dicee.knowledge_graph.KG method) function() (dicee.models.FMult2 method) (dicee.models.function_space.FMult2 method)

G

gamma (dicee.models.FMult attribute) (dicee.models.function_space.FMult attribute) gate_residual (dicee.models.literal.LiteralEmbeddings attribute), [1] gated_residual_proj (dicee.models.literal.LiteralEmbeddings attribute) gelu (dicee.models.transformers.MLP attribute) gen_test (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) gen_valid (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) generate() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) (dicee.models.transformers.BytE method) generate_queries() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) get_aswa_state_dict() (dicee.weight_averaging.ASWA method) get_bpe_head_and_relation_representation() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] get_bpe_token_representation() (dicee.abstracts.BaseInteractiveKGE method) get_callbacks() (in module dicee.trainer.dice_trainer) get_default_arguments() (in module dicee.analyse_experiments) (in module dicee.scripts.index_serve) (in module dicee.scripts.run) get_ee_vocab() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) (in module dicee.static_preprocess_funcs) get_embeddings() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] (dicee.models.ensemble.EnsembleKGE method) (dicee.models.real.Shallom method) (dicee.models.Shallom method) get_entity_embeddings() (dicee.abstracts.BaseInteractiveKGE method) get_entity_index() (dicee.abstracts.BaseInteractiveKGE method) get_er_vocab() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) (in module dicee.static_preprocess_funcs)

get_eval_report() (dicee.abstracts.BaseInteractiveKGE method) get_head_relation_representation() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] get_kronecker_triple_representation() (dicee.callbacks.KronE method) get_mean_and_var() (dicee.weight_averaging.SWAG method) get_num_params() (dicee.models.transformers.GPT method) get_padded_bpe_triple_representation() (dicee.abstracts.BaseInteractiveKGE method) get_queries() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) get_re_vocab() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) (in module dicee.static_preprocess_funcs) get_relation_embeddings() (dicee.abstracts.BaseInteractiveKGE method) get_relation_index() (dicee.abstracts.BaseInteractiveKGE method) get_sentence_representation() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] get_transductive_entity_embeddings() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) get_triple_representation() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] GFMult (class in dicee.models) (class in dicee.models.function_space) global_rank (dicee.abstracts.AbstractTrainer attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) GPT (class in dicee.models.transformers) GPTConfig (class in dicee.models.transformers) gpus (dicee.config.Namespace attribute) gradient_accumulation_steps (dicee.config.Namespace attribute) ground_queries() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) gswa_n (dicee.weight_averaging.SWAG attribute)

H

hidden_dim (dicee.models.literal.LiteralEmbeddings attribute) hidden_dropout (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] hidden_dropout_rate (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6]

hidden_normalizer (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6]

I

IdentityClass (class in dicee.models), [1], [2] (class in dicee.models.base_model) idx_entity_to_bpe_shaped (dicee.knowledge_graph.KG attribute) index() (in module dicee.scripts.index_serve) index_triple() (dicee.abstracts.BaseInteractiveKGE method) init_dataloader() (dicee.DICE_Trainer method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) init_dataset() (dicee.DICE_Trainer method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) init_param (dicee.config.Namespace attribute) init_params_with_sanity_checking() (dicee.models.base_model.BaseKGE method) (dicee.models.BaseKGE method), [1], [2], [3], [4], [5], [6] initial_eval_setting (dicee.weight_averaging.ASWA attribute) initialize_or_load_model() (dicee.DICE_Trainer method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) initialize_trainer() (dicee.DICE_Trainer method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) (in module dicee.trainer.dice_trainer)

input_dp_ent_real (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] input_dp_rel_real (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] input_dropout_rate (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] InteractiveQueryDecomposition (class in dicee.abstracts) intialize_model() (in module dicee.static_funcs) is_continual_training (dicee.DICE_Trainer attribute) (dicee.evaluation.Evaluator attribute) (dicee.evaluation.evaluator.Evaluator attribute) (dicee.Evaluator attribute) (dicee.evaluator.Evaluator attribute) (dicee.Execute attribute) (dicee.executer.Execute attribute) (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) is_global_zero (dicee.abstracts.AbstractTrainer attribute) is_rank_zero() (dicee.Execute method) (dicee.executer.Execute method) is_seen() (dicee.abstracts.BaseInteractiveKGE method) is_sparql_endpoint_alive() (in module dicee.sanity_checkers)

K

k (dicee.models.FMult attribute) (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.FMult2 attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.GFMult attribute) k_fold_cross_validation() (dicee.DICE_Trainer method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) k_vs_all_score() (dicee.models.clifford.Keci method) (dicee.models.ComplEx static method) (dicee.models.complex.ComplEx static method) (dicee.models.DistMult method) (dicee.models.Keci method) (dicee.models.octonion.OMult method) (dicee.models.OMult method) (dicee.models.QMult method) (dicee.models.quaternion.QMult method) (dicee.models.real.DistMult method) Keci (class in dicee.models) (class in dicee.models.clifford)

kernel_size (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] KG (class in dicee.knowledge_graph) kg (dicee.callbacks.PseudoLabellingCallback attribute) (dicee.read_preprocess_save_load_kg.LoadSaveToDisk attribute) (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG attribute) (dicee.read_preprocess_save_load_kg.PreprocessKG attribute) (dicee.read_preprocess_save_load_kg.read_from_disk.ReadFromDisk attribute) (dicee.read_preprocess_save_load_kg.ReadFromDisk attribute) (dicee.read_preprocess_save_load_kg.save_load_disk.LoadSaveToDisk attribute) KGE (class in dicee) (class in dicee.knowledge_graph_embeddings) KGESaveCallback (class in dicee.callbacks) knowledge_graph (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] KronE (class in dicee.callbacks) KvsAll (class in dicee.dataset_classes) kvsall_score() (dicee.models.DualE method) (dicee.models.dualE.DualE method) KvsSampleDataset (class in dicee.dataset_classes)

L

label_smoothing_rate (dicee.config.Namespace attribute) (dicee.dataset_classes.AllvsAll attribute) (dicee.dataset_classes.KvsAll attribute) (dicee.dataset_classes.KvsSampleDataset attribute) (dicee.dataset_classes.OnevsSample attribute), [1] (dicee.dataset_classes.TriplePredictionDataset attribute) labels (dicee.dataset_classes.NegSampleDataset attribute) layer_norm (dicee.models.literal.LiteralEmbeddings attribute) LayerNorm (class in dicee.models.transformers) learning_rate (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] length (dicee.dataset_classes.NegSampleDataset attribute) (dicee.dataset_classes.TriplePredictionDataset attribute) level (dicee.callbacks.Perturb attribute) LFMult (class in dicee.models) (class in dicee.models.function_space) LFMult1 (class in dicee.models) (class in dicee.models.function_space) linear() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) list2tuple() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) LiteralDataset (class in dicee.dataset_classes) LiteralEmbeddings (class in dicee.models.literal) lm_head (dicee.models.transformers.BytE attribute) (dicee.models.transformers.GPT attribute) ln_1 (dicee.models.Block attribute) (dicee.models.transformers.Block attribute) ln_2 (dicee.models.Block attribute) (dicee.models.transformers.Block attribute) ln_f (dicee.models.CoKE attribute) (dicee.models.real.CoKE attribute) load() (dicee.read_preprocess_save_load_kg.LoadSaveToDisk method) (dicee.read_preprocess_save_load_kg.save_load_disk.LoadSaveToDisk method) load_and_validate_literal_data() (dicee.dataset_classes.LiteralDataset static method) load_from_memmap() (dicee.Execute method) (dicee.executer.Execute method) load_json() (in module dicee.static_funcs) load_model() (in module dicee.static_funcs)

load_model_ensemble() (in module dicee.static_funcs) load_numpy() (in module dicee.static_funcs) load_numpy_ndarray() (in module dicee.read_preprocess_save_load_kg.util) load_pickle() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) load_queries() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) load_queries_and_answers() (dicee.query_generator.QueryGenerator static method) (dicee.QueryGenerator static method) load_state_dict() (dicee.models.ensemble.EnsembleKGE method) load_term_mapping() (in module dicee.static_funcs) (in module dicee.trainer.dice_trainer) load_with_pandas() (in module dicee.read_preprocess_save_load_kg.util) loader_backend (dicee.dataset_classes.LiteralDataset attribute), [1] LoadSaveToDisk (class in dicee.read_preprocess_save_load_kg) (class in dicee.read_preprocess_save_load_kg.save_load_disk) local_rank (dicee.abstracts.AbstractTrainer attribute) (dicee.Execute attribute) (dicee.executer.Execute attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) loss (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] loss_func (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) loss_function (dicee.trainer.torch_trainer.TorchTrainer attribute) loss_function() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) (dicee.models.transformers.BytE method) loss_history (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) lr (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) lr_init (dicee.weight_averaging.SWA attribute) (dicee.weight_averaging.SWAG attribute) (dicee.weight_averaging.TWA attribute) lr_lambda (dicee.callbacks.LRScheduler attribute) LRScheduler (class in dicee.callbacks)

M

m (dicee.models.function_space.LFMult attribute) (dicee.models.LFMult attribute) main() (in module dicee.scripts.index_serve) (in module dicee.scripts.run) make_iterable_verbose() (in module dicee.evaluation) (in module dicee.evaluation.utils) (in module dicee.static_funcs_training) (in module dicee.trainer.torch_trainer_ddp) mapping_from_first_two_cols_to_third() (in module dicee.static_preprocess_funcs) margin (dicee.models.Pyke attribute) (dicee.models.real.Pyke attribute) (dicee.models.real.TransE attribute) (dicee.models.TransE attribute) mask_emb (dicee.models.CoKE attribute) (dicee.models.real.CoKE attribute) max_ans_num (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) max_epochs (dicee.callbacks.KGESaveCallback attribute) (dicee.callbacks.PeriodicEvalCallback attribute) (dicee.weight_averaging.EMA attribute) (dicee.weight_averaging.SWA attribute) (dicee.weight_averaging.SWAG attribute) (dicee.weight_averaging.TWA attribute) max_length_subword_tokens (dicee.knowledge_graph.KG attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] max_num_models (dicee.weight_averaging.SWAG attribute) max_num_of_classes (dicee.dataset_classes.KvsSampleDataset attribute) mean (dicee.weight_averaging.SWAG attribute) mem_of_model() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) (dicee.models.ensemble.EnsembleKGE method) method (dicee.callbacks.Perturb attribute) MLP (class in dicee.models.transformers) mlp (dicee.models.Block attribute) (dicee.models.transformers.Block attribute) mode (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) model (dicee.config.Namespace attribute) (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) (dicee.trainer.torch_trainer.TorchTrainer attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) model_kwargs (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) model_name (dicee.analyse_experiments.Experiment attribute) MODEL_REGISTRY (in module dicee.static_funcs) module dicee dicee.__main__ dicee.abstracts dicee.analyse_experiments dicee.callbacks dicee.config dicee.dataset_classes dicee.eval_static_funcs dicee.evaluation dicee.evaluation.ensemble dicee.evaluation.evaluator dicee.evaluation.link_prediction dicee.evaluation.literal_prediction dicee.evaluation.utils dicee.evaluator dicee.executer dicee.knowledge_graph dicee.knowledge_graph_embeddings dicee.models dicee.models.adopt dicee.models.base_model dicee.models.clifford dicee.models.complex dicee.models.dualE dicee.models.ensemble dicee.models.function_space dicee.models.literal dicee.models.octonion dicee.models.pykeen_models dicee.models.quaternion dicee.models.real dicee.models.static_funcs dicee.models.transformers dicee.query_generator dicee.read_preprocess_save_load_kg dicee.read_preprocess_save_load_kg.preprocess dicee.read_preprocess_save_load_kg.read_from_disk dicee.read_preprocess_save_load_kg.save_load_disk dicee.read_preprocess_save_load_kg.util dicee.sanity_checkers dicee.scripts dicee.scripts.index_serve dicee.scripts.run dicee.static_funcs dicee.static_funcs_training dicee.static_preprocess_funcs dicee.trainer dicee.trainer.dice_trainer dicee.trainer.model_parallelism dicee.trainer.torch_trainer dicee.trainer.torch_trainer_ddp dicee.weight_averaging

modules() (dicee.models.ensemble.EnsembleKGE method) moving_average() (dicee.weight_averaging.SWA static method) MultiClassClassificationDataset (class in dicee.dataset_classes) MultiLabelDataset (class in dicee.dataset_classes)

N

n (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult2 attribute) n_embd (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) (dicee.models.transformers.SelfAttention attribute) n_epochs_eval_model (dicee.callbacks.PeriodicEvalCallback attribute) (dicee.config.Namespace attribute) n_head (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) (dicee.models.transformers.SelfAttention attribute) n_layer (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) n_layers (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult2 attribute) name (dicee.abstracts.BaseInteractiveKGE property) (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.AConvQ attribute) (dicee.models.CKeci attribute) (dicee.models.clifford.CKeci attribute) (dicee.models.clifford.DeCaL attribute) (dicee.models.clifford.Keci attribute) (dicee.models.CoKE attribute) (dicee.models.ComplEx attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ComplEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.ConvQ attribute) (dicee.models.DeCaL attribute) (dicee.models.DistMult attribute) (dicee.models.DualE attribute) (dicee.models.dualE.DualE attribute) (dicee.models.ensemble.EnsembleKGE attribute) (dicee.models.FMult attribute) (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.FMult2 attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.function_space.LFMult attribute) (dicee.models.function_space.LFMult1 attribute) (dicee.models.GFMult attribute) (dicee.models.Keci attribute) (dicee.models.LFMult attribute) (dicee.models.LFMult1 attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) (dicee.models.octonion.OMult attribute) (dicee.models.OMult attribute) (dicee.models.Pyke attribute) (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) (dicee.models.QMult attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute) (dicee.models.quaternion.QMult attribute) (dicee.models.real.CoKE attribute) (dicee.models.real.DistMult attribute) (dicee.models.real.Pyke attribute) (dicee.models.real.Shallom attribute) (dicee.models.real.TransE attribute) (dicee.models.Shallom attribute) (dicee.models.TransE attribute) (dicee.models.transformers.BytE attribute) named_children() (dicee.models.ensemble.EnsembleKGE method) Namespace (class in dicee.config) neg_ratio (dicee.config.Namespace attribute) (dicee.dataset_classes.BPE_NegativeSamplingDataset attribute) (dicee.dataset_classes.KvsSampleDataset attribute) neg_sample_ratio (dicee.dataset_classes.CVDataModule attribute) (dicee.dataset_classes.NegSampleDataset attribute) (dicee.dataset_classes.OnevsSample attribute), [1] (dicee.dataset_classes.TriplePredictionDataset attribute) negnorm() (dicee.abstracts.InteractiveQueryDecomposition method) NegSampleDataset (class in dicee.dataset_classes)

neural_searcher (in module dicee.scripts.index_serve) NeuralSearcher (class in dicee.scripts.index_serve) NodeTrainer (class in dicee.trainer.torch_trainer_ddp) norm_fc1 (dicee.models.AConEx attribute) (dicee.models.AConvO attribute) (dicee.models.complex.AConEx attribute) (dicee.models.complex.ConEx attribute) (dicee.models.ConEx attribute) (dicee.models.ConvO attribute) (dicee.models.octonion.AConvO attribute) (dicee.models.octonion.ConvO attribute) normalization (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) (dicee.dataset_classes.LiteralDataset attribute) normalization_params (dicee.dataset_classes.LiteralDataset attribute), [1] normalization_type (dicee.dataset_classes.LiteralDataset attribute) normalize_head_entity_embeddings (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] normalize_relation_embeddings (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] normalize_tail_entity_embeddings (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] normalizer_class (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] num_bpe_entities (dicee.dataset_classes.BPE_NegativeSamplingDataset attribute) (dicee.knowledge_graph.KG attribute) num_core (dicee.config.Namespace attribute) num_data_properties (dicee.dataset_classes.LiteralDataset attribute) num_datapoints (dicee.dataset_classes.BPE_NegativeSamplingDataset attribute) (dicee.dataset_classes.MultiLabelDataset attribute) num_ent (dicee.models.DualE attribute) (dicee.models.dualE.DualE attribute) num_entities (dicee.dataset_classes.CVDataModule attribute) (dicee.dataset_classes.KvsSampleDataset attribute) (dicee.dataset_classes.LiteralDataset attribute), [1] (dicee.dataset_classes.NegSampleDataset attribute) (dicee.dataset_classes.OnevsSample attribute), [1] (dicee.dataset_classes.TriplePredictionDataset attribute) (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] (dicee.knowledge_graph.KG attribute), [1] (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] num_epochs (dicee.abstracts.AbstractPPECallback attribute) (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) (dicee.weight_averaging.ASWA attribute) num_folds_for_cv (dicee.config.Namespace attribute) num_of_data_points (dicee.dataset_classes.MultiClassClassificationDataset attribute) num_of_data_properties (dicee.models.literal.LiteralEmbeddings attribute), [1] num_of_epochs (dicee.callbacks.PseudoLabellingCallback attribute) num_of_output_channels (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] num_params (dicee.analyse_experiments.Experiment attribute) num_relations (dicee.dataset_classes.CVDataModule attribute) (dicee.dataset_classes.NegSampleDataset attribute) (dicee.dataset_classes.OnevsSample attribute), [1] (dicee.dataset_classes.TriplePredictionDataset attribute) (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] (dicee.knowledge_graph.KG attribute), [1] (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] num_sample (dicee.models.FMult attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.GFMult attribute) num_samples (dicee.weight_averaging.TWA attribute) num_tokens (dicee.knowledge_graph.KG attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] num_workers (dicee.dataset_classes.CVDataModule attribute) numpy_data_type_changer() (in module dicee.static_funcs)

O

octonion_mul() (in module dicee.models) (in module dicee.models.octonion) octonion_mul_norm() (in module dicee.models) (in module dicee.models.octonion) octonion_normalizer() (dicee.models.AConvO static method) (dicee.models.ConvO static method) (dicee.models.octonion.AConvO static method) (dicee.models.octonion.ConvO static method) (dicee.models.octonion.OMult static method) (dicee.models.OMult static method) OMult (class in dicee.models) (class in dicee.models.octonion) on_epoch_end() (dicee.callbacks.KGESaveCallback method) (dicee.callbacks.PseudoLabellingCallback method) on_fit_end() (dicee.abstracts.AbstractCallback method) (dicee.abstracts.AbstractPPECallback method) (dicee.abstracts.AbstractTrainer method) (dicee.callbacks.AccumulateEpochLossCallback method) (dicee.callbacks.Eval method) (dicee.callbacks.KGESaveCallback method) (dicee.callbacks.LRScheduler method) (dicee.callbacks.PeriodicEvalCallback method) (dicee.callbacks.PrintCallback method) (dicee.weight_averaging.ASWA method) (dicee.weight_averaging.EMA method) (dicee.weight_averaging.SWA method) (dicee.weight_averaging.SWAG method) (dicee.weight_averaging.TWA method) on_fit_start() (dicee.abstracts.AbstractCallback method) (dicee.abstracts.AbstractPPECallback method) (dicee.abstracts.AbstractTrainer method) (dicee.callbacks.Eval method) (dicee.callbacks.KGESaveCallback method) (dicee.callbacks.KronE method) (dicee.callbacks.PrintCallback method) on_init_end() (dicee.abstracts.AbstractCallback method)

on_init_start() (dicee.abstracts.AbstractCallback method) on_train_batch_end() (dicee.abstracts.AbstractCallback method) (dicee.abstracts.AbstractTrainer method) (dicee.callbacks.Eval method) (dicee.callbacks.KGESaveCallback method) (dicee.callbacks.LRScheduler method) (dicee.callbacks.PrintCallback method) on_train_batch_start() (dicee.callbacks.Perturb method) on_train_epoch_end() (dicee.abstracts.AbstractCallback method) (dicee.abstracts.AbstractTrainer method) (dicee.callbacks.Eval method) (dicee.callbacks.KGESaveCallback method) (dicee.callbacks.PeriodicEvalCallback method) (dicee.callbacks.PrintCallback method) (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) (dicee.weight_averaging.ASWA method) (dicee.weight_averaging.EMA method) (dicee.weight_averaging.SWA method) (dicee.weight_averaging.SWAG method) (dicee.weight_averaging.TWA method) on_train_epoch_start() (dicee.abstracts.AbstractTrainer method) (dicee.weight_averaging.EMA method) (dicee.weight_averaging.SWA method) (dicee.weight_averaging.SWAG method) (dicee.weight_averaging.TWA method) on_train_start() (dicee.callbacks.LRScheduler method) OnevsAllDataset (class in dicee.dataset_classes) OnevsSample (class in dicee.dataset_classes) optim (dicee.config.Namespace attribute) optimizer (dicee.trainer.torch_trainer.TorchTrainer attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) optimizer_name (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] ordered_bpe_entities (dicee.dataset_classes.BPE_NegativeSamplingDataset attribute) (dicee.knowledge_graph.KG attribute) ordered_shaped_bpe_tokens (dicee.knowledge_graph.KG attribute)

P

p (dicee.config.Namespace attribute) (dicee.models.clifford.DeCaL attribute) (dicee.models.clifford.Keci attribute) (dicee.models.DeCaL attribute) (dicee.models.Keci attribute) P (dicee.weight_averaging.TWA attribute) padding (dicee.knowledge_graph.KG attribute) pandas_dataframe_indexer() (in module dicee.read_preprocess_save_load_kg.util) param_init (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] parameters() (dicee.abstracts.BaseInteractiveKGE method) (dicee.models.ensemble.EnsembleKGE method) path (dicee.abstracts.AbstractPPECallback attribute) (dicee.callbacks.AccumulateEpochLossCallback attribute) (dicee.callbacks.Eval attribute) (dicee.callbacks.KGESaveCallback attribute) (dicee.weight_averaging.ASWA attribute) path_dataset_folder (dicee.analyse_experiments.Experiment attribute) path_for_deserialization (dicee.knowledge_graph.KG attribute) path_for_serialization (dicee.knowledge_graph.KG attribute) path_single_kg (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) path_to_store_single_run (dicee.config.Namespace attribute) PeriodicEvalCallback (class in dicee.callbacks) Perturb (class in dicee.callbacks) polars_dataframe_indexer() (in module dicee.read_preprocess_save_load_kg.util) poly_NN() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) polynomial() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) pop() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) pos_emb (dicee.models.CoKE attribute) (dicee.models.real.CoKE attribute)

pq (dicee.analyse_experiments.Experiment attribute) predict() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) predict_dataloader() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) predict_literals() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) predict_missing_head_entity() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) predict_missing_relations() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) predict_missing_tail_entity() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) predict_topk() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) prepare_data() (dicee.dataset_classes.CVDataModule method) preprocess_with_byte_pair_encoding() (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG method) (dicee.read_preprocess_save_load_kg.PreprocessKG method) preprocess_with_byte_pair_encoding_with_padding() (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG method) (dicee.read_preprocess_save_load_kg.PreprocessKG method) preprocess_with_pandas() (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG method) (dicee.read_preprocess_save_load_kg.PreprocessKG method) preprocess_with_polars() (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG method) (dicee.read_preprocess_save_load_kg.PreprocessKG method) preprocesses_input_args() (in module dicee.static_preprocess_funcs) PreprocessKG (class in dicee.read_preprocess_save_load_kg) (class in dicee.read_preprocess_save_load_kg.preprocess) PrintCallback (class in dicee.callbacks) process (dicee.trainer.torch_trainer.TorchTrainer attribute) PseudoLabellingCallback (class in dicee.callbacks) Pyke (class in dicee.models) (class in dicee.models.real) pykeen_model_kwargs (dicee.config.Namespace attribute) PykeenKGE (class in dicee.models) (class in dicee.models.pykeen_models)

Q

q (dicee.config.Namespace attribute) (dicee.models.clifford.DeCaL attribute) (dicee.models.clifford.Keci attribute) (dicee.models.DeCaL attribute) (dicee.models.Keci attribute) qdrant_client (dicee.scripts.index_serve.NeuralSearcher attribute) QMult (class in dicee.models) (class in dicee.models.quaternion) quaternion_mul() (in module dicee.models) (in module dicee.models.static_funcs)

quaternion_mul_with_unit_norm() (in module dicee.models) (in module dicee.models.quaternion) quaternion_multiplication_followed_by_inner_product() (dicee.models.QMult method) (dicee.models.quaternion.QMult method) quaternion_normalizer() (dicee.models.QMult static method) (dicee.models.quaternion.QMult static method) queries (dicee.scripts.index_serve.StringListRequest attribute) query_name_to_struct (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) QueryGenerator (class in dicee) (class in dicee.query_generator)

R

r (dicee.models.clifford.DeCaL attribute) (dicee.models.clifford.Keci attribute) (dicee.models.DeCaL attribute) (dicee.models.Keci attribute) random_seed (dicee.config.Namespace attribute) range_constraints_per_rel (dicee.evaluation.Evaluator attribute) (dicee.evaluation.evaluator.Evaluator attribute) (dicee.Evaluator attribute) (dicee.evaluator.Evaluator attribute) rank (dicee.Execute attribute) (dicee.executer.Execute attribute) ratio (dicee.callbacks.Perturb attribute) raw_test_set (dicee.knowledge_graph.KG attribute) raw_train_set (dicee.knowledge_graph.KG attribute) raw_valid_set (dicee.knowledge_graph.KG attribute) re (dicee.models.clifford.DeCaL attribute) (dicee.models.DeCaL attribute) re_vocab (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] read_from_disk() (in module dicee.read_preprocess_save_load_kg.util) read_from_triple_store_with_pandas() (in module dicee.read_preprocess_save_load_kg.util) read_from_triple_store_with_polars() (in module dicee.read_preprocess_save_load_kg.util) read_only_few (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) read_or_load_kg() (in module dicee.static_funcs) read_with_pandas() (in module dicee.read_preprocess_save_load_kg.util) read_with_polars() (in module dicee.read_preprocess_save_load_kg.util) ReadFromDisk (class in dicee.read_preprocess_save_load_kg) (class in dicee.read_preprocess_save_load_kg.read_from_disk) reducer (dicee.scripts.index_serve.StringListRequest attribute) reg_lambda (dicee.weight_averaging.TWA attribute) rel2id (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) relation_embeddings (dicee.models.AConvQ attribute) (dicee.models.clifford.DeCaL attribute) (dicee.models.ConvQ attribute) (dicee.models.DeCaL attribute) (dicee.models.DualE attribute) (dicee.models.dualE.DualE attribute) (dicee.models.FMult attribute) (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.FMult2 attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.function_space.LFMult attribute) (dicee.models.function_space.LFMult1 attribute) (dicee.models.GFMult attribute) (dicee.models.LFMult attribute) (dicee.models.LFMult1 attribute) (dicee.models.pykeen_models.PykeenKGE attribute) (dicee.models.PykeenKGE attribute) (dicee.models.quaternion.AConvQ attribute) (dicee.models.quaternion.ConvQ attribute)

relation_to_idx (dicee.knowledge_graph.KG attribute), [1] relations_str (dicee.knowledge_graph.KG property) reload_dataset() (in module dicee.dataset_classes) report (dicee.DICE_Trainer attribute) (dicee.evaluation.Evaluator attribute), [1] (dicee.evaluation.evaluator.Evaluator attribute), [1] (dicee.Evaluator attribute), [1] (dicee.evaluator.Evaluator attribute), [1] (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) reports (dicee.callbacks.Eval attribute) (dicee.callbacks.PeriodicEvalCallback attribute) requires_grad_for_interactions (dicee.models.CKeci attribute) (dicee.models.clifford.CKeci attribute) (dicee.models.clifford.Keci attribute) (dicee.models.Keci attribute) resid_dropout (dicee.models.transformers.SelfAttention attribute) residual_convolution() (dicee.models.AConEx method) (dicee.models.AConvO method) (dicee.models.AConvQ method) (dicee.models.complex.AConEx method) (dicee.models.complex.ConEx method) (dicee.models.ConEx method) (dicee.models.ConvO method) (dicee.models.ConvQ method) (dicee.models.octonion.AConvO method) (dicee.models.octonion.ConvO method) (dicee.models.quaternion.AConvQ method) (dicee.models.quaternion.ConvQ method) retrieve_embedding() (dicee.scripts.index_serve.NeuralSearcher method) retrieve_embeddings() (in module dicee.scripts.index_serve) return_multi_hop_query_results() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) root() (in module dicee.scripts.index_serve) roots (dicee.models.FMult attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.GFMult attribute) runtime (dicee.analyse_experiments.Experiment attribute)

S

sample() (dicee.weight_averaging.SWAG method) sample_counter (dicee.abstracts.AbstractPPECallback attribute) sample_entity() (dicee.abstracts.BaseInteractiveKGE method) sample_relation() (dicee.abstracts.BaseInteractiveKGE method) sample_triples_ratio (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) sample_weights() (dicee.weight_averaging.TWA method) sampling_ratio (dicee.dataset_classes.LiteralDataset attribute), [1] sanity_check_callback_args() (in module dicee.sanity_checkers) sanity_checking_with_arguments() (in module dicee.sanity_checkers) save() (dicee.abstracts.BaseInteractiveKGE method) (dicee.read_preprocess_save_load_kg.LoadSaveToDisk method) (dicee.read_preprocess_save_load_kg.save_load_disk.LoadSaveToDisk method) save_checkpoint() (dicee.abstracts.AbstractTrainer static method) save_checkpoint_model() (in module dicee.static_funcs) save_embeddings() (in module dicee.static_funcs) save_embeddings_as_csv (dicee.config.Namespace attribute) save_every_n_epochs (dicee.config.Namespace attribute) save_experiment() (dicee.analyse_experiments.Experiment method) save_model_at_every_epoch (dicee.config.Namespace attribute) save_model_every_n_epoch (dicee.callbacks.PeriodicEvalCallback attribute) save_numpy_ndarray() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) save_pickle() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) save_queries() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) save_queries_and_answers() (dicee.query_generator.QueryGenerator static method) (dicee.QueryGenerator static method) save_trained_model() (dicee.Execute method) (dicee.executer.Execute method) scalar_batch_NN() (dicee.models.function_space.LFMult method) (dicee.models.LFMult method) scaler (dicee.callbacks.Perturb attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) scheduler (dicee.callbacks.LRScheduler attribute) score() (dicee.models.clifford.Keci method) (dicee.models.CoKE method) (dicee.models.ComplEx static method) (dicee.models.complex.ComplEx static method) (dicee.models.DistMult method) (dicee.models.Keci method) (dicee.models.octonion.OMult method) (dicee.models.OMult method) (dicee.models.QMult method) (dicee.models.quaternion.QMult method) (dicee.models.real.CoKE method) (dicee.models.real.DistMult method) (dicee.models.real.TransE method) (dicee.models.TransE method) score_func (dicee.models.FMult2 attribute) (dicee.models.function_space.FMult2 attribute) scoring_technique (dicee.analyse_experiments.Experiment attribute) (dicee.config.Namespace attribute) search() (dicee.scripts.index_serve.NeuralSearcher method) search_embeddings() (in module dicee.scripts.index_serve) search_embeddings_batch() (in module dicee.scripts.index_serve) seed (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) select_model() (in module dicee.static_funcs) selected_optimizer (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6]

SelfAttention (class in dicee.models.transformers) separator (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) sequential_vocabulary_construction() (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG method) (dicee.read_preprocess_save_load_kg.PreprocessKG method) serve() (in module dicee.scripts.index_serve) set_global_seed() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) set_model_eval_mode() (dicee.abstracts.BaseInteractiveKGE method) set_model_train_mode() (dicee.abstracts.BaseInteractiveKGE method) setup() (dicee.dataset_classes.CVDataModule method) setup_executor() (dicee.Execute method) (dicee.executer.Execute method) Shallom (class in dicee.models) (class in dicee.models.real) shallom (dicee.models.real.Shallom attribute) (dicee.models.Shallom attribute) single_hop_query_answering() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) snapshot_dir (dicee.callbacks.LRScheduler attribute) snapshot_loss (dicee.callbacks.LRScheduler attribute) sparql_endpoint (dicee.config.Namespace attribute) (dicee.knowledge_graph.KG attribute) sq_mean (dicee.weight_averaging.SWAG attribute) start() (dicee.DICE_Trainer method) (dicee.Execute method) (dicee.executer.Execute method) (dicee.read_preprocess_save_load_kg.preprocess.PreprocessKG method) (dicee.read_preprocess_save_load_kg.PreprocessKG method) (dicee.read_preprocess_save_load_kg.read_from_disk.ReadFromDisk method) (dicee.read_preprocess_save_load_kg.ReadFromDisk method) (dicee.trainer.DICE_Trainer method) (dicee.trainer.dice_trainer.DICE_Trainer method) start_time (dicee.callbacks.PrintCallback attribute) (dicee.Execute attribute) (dicee.executer.Execute attribute) state_dict() (dicee.models.ensemble.EnsembleKGE method) step() (dicee.models.ADOPT method) (dicee.models.adopt.ADOPT method) (dicee.models.ensemble.EnsembleKGE method) step_count (dicee.callbacks.LRScheduler attribute) storage_path (dicee.config.Namespace attribute) (dicee.DICE_Trainer attribute) (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) store() (in module dicee.static_funcs) store_ensemble() (dicee.abstracts.AbstractPPECallback method) strategy (dicee.abstracts.AbstractTrainer attribute) StringListRequest (class in dicee.scripts.index_serve) SWA (class in dicee.weight_averaging) swa (dicee.config.Namespace attribute) swa_c_epochs (dicee.config.Namespace attribute) (dicee.weight_averaging.SWA attribute) (dicee.weight_averaging.SWAG attribute) swa_lr (dicee.weight_averaging.SWA attribute) (dicee.weight_averaging.SWAG attribute) swa_model (dicee.weight_averaging.SWA attribute) swa_n (dicee.weight_averaging.SWA attribute) swa_start_epoch (dicee.config.Namespace attribute) (dicee.weight_averaging.SWA attribute) (dicee.weight_averaging.SWAG attribute) SWAG (class in dicee.weight_averaging) swag (dicee.config.Namespace attribute)

T

T() (dicee.models.DualE method) (dicee.models.dualE.DualE method) t_conorm() (dicee.abstracts.InteractiveQueryDecomposition method) t_norm() (dicee.abstracts.InteractiveQueryDecomposition method) target_dim (dicee.dataset_classes.AllvsAll attribute) (dicee.dataset_classes.MultiLabelDataset attribute) (dicee.dataset_classes.OnevsAllDataset attribute) (dicee.knowledge_graph.KG attribute) temperature (dicee.models.transformers.BytE attribute) tensor_t_norm() (dicee.abstracts.InteractiveQueryDecomposition method) TensorParallel (class in dicee.trainer.model_parallelism) test_dataloader() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) test_epoch_end() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) test_h1 (dicee.analyse_experiments.Experiment attribute) test_h10 (dicee.analyse_experiments.Experiment attribute) test_h3 (dicee.analyse_experiments.Experiment attribute) test_mrr (dicee.analyse_experiments.Experiment attribute) test_path (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) test_set (dicee.knowledge_graph.KG attribute), [1] timeit() (in module dicee.read_preprocess_save_load_kg.util) (in module dicee.static_funcs) (in module dicee.static_preprocess_funcs) to() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) (dicee.models.ensemble.EnsembleKGE method) to_df() (dicee.analyse_experiments.Experiment method) topk (dicee.models.transformers.BytE attribute) (dicee.scripts.index_serve.NeuralSearcher attribute) torch_ordered_shaped_bpe_entities (dicee.dataset_classes.MultiLabelDataset attribute) TorchDDPTrainer (class in dicee.trainer.torch_trainer_ddp) TorchTrainer (class in dicee.trainer.torch_trainer) total_epochs (dicee.callbacks.LRScheduler attribute) total_steps (dicee.callbacks.LRScheduler attribute) train() (dicee.abstracts.BaseInteractiveTrainKGE method) (dicee.trainer.torch_trainer_ddp.NodeTrainer method) train_data (dicee.dataset_classes.AllvsAll attribute) (dicee.dataset_classes.KvsAll attribute) (dicee.dataset_classes.KvsSampleDataset attribute) (dicee.dataset_classes.MultiClassClassificationDataset attribute) (dicee.dataset_classes.OnevsAllDataset attribute) (dicee.dataset_classes.OnevsSample attribute), [1] train_dataloader() (dicee.dataset_classes.CVDataModule method) (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) train_dataloaders (dicee.trainer.torch_trainer.TorchTrainer attribute) train_dataset_loader (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) train_file_path (dicee.dataset_classes.LiteralDataset attribute), [1] train_h1 (dicee.analyse_experiments.Experiment attribute) train_h10 (dicee.analyse_experiments.Experiment attribute) train_h3 (dicee.analyse_experiments.Experiment attribute) train_indices_target (dicee.dataset_classes.MultiLabelDataset attribute) train_k_vs_all() (dicee.abstracts.BaseInteractiveTrainKGE method)

train_literals() (dicee.abstracts.BaseInteractiveTrainKGE method) train_mode (dicee.models.ensemble.EnsembleKGE attribute) train_mrr (dicee.analyse_experiments.Experiment attribute) train_path (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) train_set (dicee.dataset_classes.BPE_NegativeSamplingDataset attribute) (dicee.dataset_classes.MultiLabelDataset attribute) (dicee.dataset_classes.NegSampleDataset attribute) (dicee.dataset_classes.TriplePredictionDataset attribute) (dicee.knowledge_graph.KG attribute), [1] train_set_idx (dicee.dataset_classes.CVDataModule attribute) train_set_target (dicee.knowledge_graph.KG attribute) train_target (dicee.dataset_classes.AllvsAll attribute) (dicee.dataset_classes.KvsAll attribute) (dicee.dataset_classes.KvsSampleDataset attribute) train_target_indices (dicee.knowledge_graph.KG attribute) train_triples (dicee.dataset_classes.NegSampleDataset attribute) train_triples() (dicee.abstracts.BaseInteractiveTrainKGE method) trained_model (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] trainer (dicee.config.Namespace attribute) (dicee.DICE_Trainer attribute) (dicee.Execute attribute), [1] (dicee.executer.Execute attribute), [1] (dicee.trainer.DICE_Trainer attribute) (dicee.trainer.dice_trainer.DICE_Trainer attribute) (dicee.trainer.torch_trainer_ddp.NodeTrainer attribute) training_step (dicee.trainer.torch_trainer.TorchTrainer attribute) training_step() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) (dicee.models.transformers.BytE method) training_step_outputs (dicee.models.base_model.BaseKGELightning attribute) (dicee.models.BaseKGELightning attribute) training_technique (dicee.knowledge_graph.KG attribute) TransE (class in dicee.models) (class in dicee.models.real) transfer_batch_to_device() (dicee.dataset_classes.CVDataModule method) transformer (dicee.models.transformers.BytE attribute) (dicee.models.transformers.GPT attribute) trapezoid() (dicee.models.FMult2 method) (dicee.models.function_space.FMult2 method) tri_score() (dicee.models.function_space.LFMult method) (dicee.models.function_space.LFMult1 method) (dicee.models.LFMult method) (dicee.models.LFMult1 method) triple_score() (dicee.KGE method) (dicee.knowledge_graph_embeddings.KGE method) TriplePredictionDataset (class in dicee.dataset_classes) tuple2list() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) TWA (class in dicee.weight_averaging) twa (dicee.config.Namespace attribute) twa_c_epochs (dicee.weight_averaging.TWA attribute) twa_model (dicee.weight_averaging.TWA attribute) twa_start_epoch (dicee.weight_averaging.TWA attribute)

U

unlabelled_size (dicee.callbacks.PseudoLabellingCallback attribute) unmap() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method)

unmap_query() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) update_hits() (in module dicee.evaluation.utils)

V

val_aswa (dicee.weight_averaging.ASWA attribute) val_dataloader() (dicee.models.base_model.BaseKGELightning method) (dicee.models.BaseKGELightning method) val_h1 (dicee.analyse_experiments.Experiment attribute) val_h10 (dicee.analyse_experiments.Experiment attribute) val_h3 (dicee.analyse_experiments.Experiment attribute) val_mrr (dicee.analyse_experiments.Experiment attribute) val_path (dicee.query_generator.QueryGenerator attribute) (dicee.QueryGenerator attribute) valid_set (dicee.knowledge_graph.KG attribute), [1] validate_knowledge_graph() (in module dicee.sanity_checkers) var_clamp (dicee.weight_averaging.SWAG attribute)

vocab_preparation() (dicee.evaluation.Evaluator method) (dicee.evaluation.evaluator.Evaluator method) (dicee.Evaluator method) (dicee.evaluator.Evaluator method) vocab_size (dicee.models.CoKEConfig attribute), [1] (dicee.models.real.CoKEConfig attribute), [1] (dicee.models.transformers.GPTConfig attribute) vocab_to_parquet() (in module dicee.static_funcs) vtp_score() (dicee.models.function_space.LFMult method) (dicee.models.function_space.LFMult1 method) (dicee.models.LFMult method) (dicee.models.LFMult1 method)

W

warmup_steps (dicee.callbacks.LRScheduler attribute) weight (dicee.models.transformers.LayerNorm attribute) weight_decay (dicee.config.Namespace attribute) (dicee.models.base_model.BaseKGE attribute) (dicee.models.BaseKGE attribute), [1], [2], [3], [4], [5], [6] weight_samples (dicee.weight_averaging.TWA attribute) weights (dicee.models.FMult attribute) (dicee.models.function_space.FMult attribute) (dicee.models.function_space.GFMult attribute) (dicee.models.GFMult attribute)

world_size (dicee.Execute attribute) (dicee.executer.Execute attribute) write_csv_from_model_parallel() (in module dicee.static_funcs) write_links() (dicee.query_generator.QueryGenerator method) (dicee.QueryGenerator method) write_report() (dicee.Execute method) (dicee.executer.Execute method)

X

x_values (dicee.models.function_space.LFMult attribute) (dicee.models.LFMult attribute)