Index

A

AcrE (class in pykg2vec.models.projection)

ANALOGY (class in pykg2vec.models.pointwise)

B

BaysOptimizer (class in pykg2vec.utils.bayesian_optimizer)

build_model() (pykg2vec.utils.trainer.Trainer method)

C

Complex (class in pykg2vec.models.pointwise) ComplexN3 (class in pykg2vec.models.pointwise) Config (class in pykg2vec.config) ConvE (class in pykg2vec.models.projection)

ConvKB (class in pykg2vec.models.pointwise) CP (class in pykg2vec.models.pointwise) create_feeder_process() (pykg2vec.data.generator.Generator method) create_train_processor_process() (pykg2vec.data.generator.Generator method) Criterion (class in pykg2vec.utils.criterion)

D

dataset (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) dataset_home_path (pykg2vec.data.datasets.KnownDataset attribute) (pykg2vec.data.datasets.UserDefinedDataset attribute) dataset_name (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) DeepLearning50a (class in pykg2vec.data.datasets) display() (pykg2vec.utils.trainer.Trainer method) display_summary() (pykg2vec.utils.evaluator.MetricCalculator method)

DistMult (class in pykg2vec.models.pointwise) download() (pykg2vec.data.datasets.KnownDataset method) draw_embedding() (pykg2vec.utils.visualization.Visualization static method) draw_embedding_rel_space() (pykg2vec.utils.visualization.Visualization static method) dump() (pykg2vec.data.datasets.KnownDataset method) (pykg2vec.data.datasets.UserDefinedDataset method) (pykg2vec.data.kgcontroller.KnowledgeGraph method)

E

EarlyStopper (class in pykg2vec.utils.trainer) embed() (pykg2vec.models.KGMeta.Model method) (pykg2vec.models.pairwise.HoLE method) (pykg2vec.models.pairwise.KG2E method) (pykg2vec.models.pairwise.NTN method) (pykg2vec.models.pairwise.Rescal method) (pykg2vec.models.pairwise.RotatE method) (pykg2vec.models.pairwise.SLM method) (pykg2vec.models.pairwise.SME method) (pykg2vec.models.pairwise.TransD method) (pykg2vec.models.pairwise.TransE method) (pykg2vec.models.pairwise.TransH method) (pykg2vec.models.pairwise.TransM method) (pykg2vec.models.pairwise.TransR method) (pykg2vec.models.pointwise.ANALOGY method) (pykg2vec.models.pointwise.Complex method) (pykg2vec.models.pointwise.ConvKB method) (pykg2vec.models.pointwise.CP method) (pykg2vec.models.pointwise.DistMult method) (pykg2vec.models.pointwise.MuRP method) (pykg2vec.models.pointwise.OctonionE method) (pykg2vec.models.pointwise.QuatE method) (pykg2vec.models.pointwise.SimplE method) (pykg2vec.models.pointwise.SimplE_ignr method) (pykg2vec.models.projection.AcrE method) (pykg2vec.models.projection.ConvE method) (pykg2vec.models.projection.HypER method) (pykg2vec.models.projection.InteractE method)

embed_complex() (pykg2vec.models.pointwise.ANALOGY method) entities (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) entity2idx (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) Evaluator (class in pykg2vec.utils.evaluator) export_embeddings() (pykg2vec.utils.trainer.Trainer method) extract() (pykg2vec.data.datasets.KnownDataset method) extract_tar() (in module pykg2vec.data.datasets) extract_zip() (in module pykg2vec.data.datasets)

F

f1() (pykg2vec.models.projection.ProjE_pointwise method) f2() (pykg2vec.models.projection.ProjE_pointwise method) forward() (pykg2vec.models.KGMeta.Model method) (pykg2vec.models.pairwise.HoLE method) (pykg2vec.models.pairwise.KG2E method) (pykg2vec.models.pairwise.NTN method) (pykg2vec.models.pairwise.Rescal method) (pykg2vec.models.pairwise.RotatE method) (pykg2vec.models.pairwise.SLM method) (pykg2vec.models.pairwise.SME method) (pykg2vec.models.pairwise.SME_BL method) (pykg2vec.models.pairwise.TransD method) (pykg2vec.models.pairwise.TransE method) (pykg2vec.models.pairwise.TransH method) (pykg2vec.models.pairwise.TransM method) (pykg2vec.models.pairwise.TransR method) (pykg2vec.models.pointwise.ANALOGY method) (pykg2vec.models.pointwise.Complex method) (pykg2vec.models.pointwise.ConvKB method) (pykg2vec.models.pointwise.CP method) (pykg2vec.models.pointwise.DistMult method) (pykg2vec.models.pointwise.MuRP method) (pykg2vec.models.pointwise.OctonionE method) (pykg2vec.models.pointwise.QuatE method) (pykg2vec.models.pointwise.SimplE method) (pykg2vec.models.pointwise.SimplE_ignr method) (pykg2vec.models.projection.AcrE method) (pykg2vec.models.projection.ConvE method) (pykg2vec.models.projection.HypER method) (pykg2vec.models.projection.InteractE method) (pykg2vec.models.projection.ProjE_pointwise method) (pykg2vec.models.projection.TuckER method)

FreebaseFB15k (class in pykg2vec.data.datasets) FreebaseFB15k_237 (class in pykg2vec.data.datasets)

G

g() (pykg2vec.models.projection.ProjE_pointwise static method) Generator (class in pykg2vec.data.generator) get_head_rank() (pykg2vec.utils.evaluator.MetricCalculator method) get_idx_n_emb() (pykg2vec.utils.visualization.Visualization method) get_reg() (pykg2vec.models.KGMeta.Model method) (pykg2vec.models.pairwise.NTN method) (pykg2vec.models.pointwise.ANALOGY method) (pykg2vec.models.pointwise.Complex method) (pykg2vec.models.pointwise.ComplexN3 method) (pykg2vec.models.pointwise.CP method) (pykg2vec.models.pointwise.DistMult method) (pykg2vec.models.pointwise.OctonionE method) (pykg2vec.models.pointwise.QuatE method) (pykg2vec.models.pointwise.SimplE method) (pykg2vec.models.projection.ProjE_pointwise method)

get_tail_rank() (pykg2vec.utils.evaluator.MetricCalculator method)

H

HoLE (class in pykg2vec.models.pairwise) hr_t (pykg2vec.data.kgcontroller.KnowledgeGraph attribute)

hr_t_train (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) HypER (class in pykg2vec.models.projection)

I

idx2entity (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) idx2relation (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) inner_forward() (pykg2vec.models.projection.ConvE method)

InteractE (class in pykg2vec.models.projection) is_cache_exists() (pykg2vec.data.kgcontroller.KnowledgeGraph method) is_meta_cache_exists() (pykg2vec.data.datasets.KnownDataset method) (pykg2vec.data.datasets.UserDefinedDataset method)

K

KG2E (class in pykg2vec.models.pairwise) kg_meta (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) KGMetaData (class in pykg2vec.data.kgcontroller)

Kinship (class in pykg2vec.data.datasets) KnowledgeGraph (class in pykg2vec.data.kgcontroller) KnownDataset (class in pykg2vec.data.datasets)

L

layer() (pykg2vec.models.pairwise.SLM method) load_model() (pykg2vec.utils.trainer.Trainer method)

load_params() (pykg2vec.models.KGMeta.Model method) Logger (class in pykg2vec.utils.logger)

M

MetricCalculator (class in pykg2vec.utils.evaluator) Model (class in pykg2vec.models.KGMeta) module pykg2vec.config pykg2vec.data.datasets pykg2vec.data.generator pykg2vec.data.kgcontroller pykg2vec.hyperparams pykg2vec.models.Domain pykg2vec.models.KGMeta pykg2vec.models.pairwise pykg2vec.models.pointwise pykg2vec.models.projection pykg2vec.test.test_generator pykg2vec.test.test_hp_loader pykg2vec.test.test_inference pykg2vec.test.test_kg pykg2vec.test.test_logger pykg2vec.test.test_model pykg2vec.test.test_trainer pykg2vec.test.test_tune_model pykg2vec.test.test_visualization pykg2vec.utils.bayesian_optimizer pykg2vec.utils.criterion pykg2vec.utils.evaluator pykg2vec.utils.logger pykg2vec.utils.riemannian_optimizer pykg2vec.utils.trainer pykg2vec.utils.visualization

MuRP (class in pykg2vec.models.pointwise)

N

name (pykg2vec.data.datasets.DeepLearning50a attribute) (pykg2vec.data.datasets.FreebaseFB15k attribute) (pykg2vec.data.datasets.FreebaseFB15k_237 attribute) (pykg2vec.data.datasets.Kinship attribute) (pykg2vec.data.datasets.Nations attribute) (pykg2vec.data.datasets.NELL_995 attribute) (pykg2vec.data.datasets.UMLS attribute) (pykg2vec.data.datasets.WordNet18 attribute) (pykg2vec.data.datasets.WordNet18_RR attribute) (pykg2vec.data.datasets.YAGO3_10 attribute)

NamedEmbedding (class in pykg2vec.models.Domain) Nations (class in pykg2vec.data.datasets) NELL_995 (class in pykg2vec.data.datasets) NTN (class in pykg2vec.models.pairwise)

O

OctonionE (class in pykg2vec.models.pointwise)

optimize() (pykg2vec.utils.bayesian_optimizer.BaysOptimizer method)

P

PairwiseModel (class in pykg2vec.models.KGMeta) plot_embedding() (pykg2vec.utils.visualization.Visualization method) plot_test_result() (pykg2vec.utils.visualization.Visualization method) plot_train_result() (pykg2vec.utils.visualization.Visualization method) PointwiseModel (class in pykg2vec.models.KGMeta) prefix (pykg2vec.data.datasets.DeepLearning50a attribute) (pykg2vec.data.datasets.FreebaseFB15k attribute) (pykg2vec.data.datasets.FreebaseFB15k_237 attribute) (pykg2vec.data.datasets.Kinship attribute) (pykg2vec.data.datasets.Nations attribute) (pykg2vec.data.datasets.NELL_995 attribute) (pykg2vec.data.datasets.UMLS attribute) (pykg2vec.data.datasets.WordNet18 attribute) (pykg2vec.data.datasets.WordNet18_RR attribute) (pykg2vec.data.datasets.YAGO3_10 attribute) prepare_data() (pykg2vec.data.kgcontroller.KnowledgeGraph method) process_function_multiclass() (in module pykg2vec.data.generator) process_function_pairwise() (in module pykg2vec.data.generator) process_function_pointwise() (in module pykg2vec.data.generator) ProjE_pointwise (class in pykg2vec.models.projection) ProjectionModel (class in pykg2vec.models.KGMeta) pykg2vec.config module pykg2vec.data.datasets module pykg2vec.data.generator module pykg2vec.data.kgcontroller module pykg2vec.hyperparams module pykg2vec.models.Domain module pykg2vec.models.KGMeta module pykg2vec.models.pairwise module

pykg2vec.models.pointwise module pykg2vec.models.projection module pykg2vec.test.test_generator module pykg2vec.test.test_hp_loader module pykg2vec.test.test_inference module pykg2vec.test.test_kg module pykg2vec.test.test_logger module pykg2vec.test.test_model module pykg2vec.test.test_trainer module pykg2vec.test.test_tune_model module pykg2vec.test.test_visualization module pykg2vec.utils.bayesian_optimizer module pykg2vec.utils.criterion module pykg2vec.utils.evaluator module pykg2vec.utils.logger module pykg2vec.utils.riemannian_optimizer module pykg2vec.utils.trainer module pykg2vec.utils.visualization module

Q

QuatE (class in pykg2vec.models.pointwise)

R

raw_data_generator() (in module pykg2vec.data.generator) read_cache_data() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_entities() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_hr_t() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_hr_t_train() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_hr_t_valid() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_mappings() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_metadata() (pykg2vec.data.datasets.KnownDataset method) (pykg2vec.data.datasets.UserDefinedDataset method) read_relation_property() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_relations() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_tr_h() (pykg2vec.data.kgcontroller.KnowledgeGraph method)

read_tr_h_train() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_tr_h_valid() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_triple_ids() (pykg2vec.data.kgcontroller.KnowledgeGraph method) read_triplets() (pykg2vec.data.kgcontroller.KnowledgeGraph method) relation2idx (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) relation_property (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) relations (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) Rescal (class in pykg2vec.models.pairwise) return_best() (pykg2vec.utils.bayesian_optimizer.BaysOptimizer method) RiemannianOptimizer (class in pykg2vec.utils.riemannian_optimizer) root_oath (pykg2vec.data.datasets.KnownDataset attribute) (pykg2vec.data.datasets.UserDefinedDataset attribute) RotatE (class in pykg2vec.models.pairwise)

S

save_model() (pykg2vec.utils.trainer.Trainer method) save_test_summary() (pykg2vec.utils.evaluator.MetricCalculator method) save_training_result() (pykg2vec.utils.trainer.Trainer method) set_ids() (pykg2vec.data.kgcontroller.Triple method) SimplE (class in pykg2vec.models.pointwise) SimplE_ignr (class in pykg2vec.models.pointwise)

Singleton (class in pykg2vec.utils.logger) SLM (class in pykg2vec.models.pairwise) SME (class in pykg2vec.models.pairwise) SME_BL (class in pykg2vec.models.pairwise) step() (pykg2vec.utils.riemannian_optimizer.RiemannianOptimizer method) stop() (pykg2vec.data.generator.Generator method) summary() (pykg2vec.config.Config method)

T

test_generator_pairwise() (in module pykg2vec.test.test_generator) test_generator_pointwise() (in module pykg2vec.test.test_generator) test_generator_projection() (in module pykg2vec.test.test_generator) test_inference() (in module pykg2vec.test.test_inference) test_kge_methods() (in module pykg2vec.test.test_model) test_known_datasets() (in module pykg2vec.test.test_kg) test_return_empty_before_optimization() (in module pykg2vec.test.test_tune_model) test_tuning() (in module pykg2vec.test.test_tune_model) testing_function() (in module pykg2vec.test.test_model) testing_function_with_args() (in module pykg2vec.test.test_inference) tr_h (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) tr_h_train (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) train_layer() (pykg2vec.models.pairwise.NTN method)

train_model() (pykg2vec.utils.trainer.Trainer method) train_model_epoch() (pykg2vec.utils.trainer.Trainer method) Trainer (class in pykg2vec.utils.trainer) TransD (class in pykg2vec.models.pairwise) TransE (class in pykg2vec.models.pairwise) TransH (class in pykg2vec.models.pairwise) TransM (class in pykg2vec.models.pairwise) TransR (class in pykg2vec.models.pairwise) Triple (class in pykg2vec.data.kgcontroller) triplets (pykg2vec.data.kgcontroller.KnowledgeGraph attribute) TuckER (class in pykg2vec.models.projection) tune_model() (pykg2vec.utils.trainer.Trainer method) tunning_function() (in module pykg2vec.test.test_tune_model)

U

UMLS (class in pykg2vec.data.datasets) url (pykg2vec.data.datasets.DeepLearning50a attribute) (pykg2vec.data.datasets.FreebaseFB15k attribute) (pykg2vec.data.datasets.FreebaseFB15k_237 attribute) (pykg2vec.data.datasets.Kinship attribute) (pykg2vec.data.datasets.Nations attribute) (pykg2vec.data.datasets.NELL_995 attribute) (pykg2vec.data.datasets.UMLS attribute) (pykg2vec.data.datasets.WordNet18 attribute) (pykg2vec.data.datasets.WordNet18_RR attribute) (pykg2vec.data.datasets.YAGO3_10 attribute)

UserDefinedDataset (class in pykg2vec.data.datasets)

V

Visualization (class in pykg2vec.utils.visualization)

W

WordNet18 (class in pykg2vec.data.datasets)

WordNet18_RR (class in pykg2vec.data.datasets)

Y

YAGO3_10 (class in pykg2vec.data.datasets)