latest
Tutorials:
Model Training
Model Evaluation
Reference:
Models
Evaluation
Inference
Negative Sampling
Data Structure
Utils
Installation:
Stable release
From sources
About:
Contributing
Credits
History
TorchKGE
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
P
|
R
|
S
|
T
|
U
A
accuracy() (torchkge.evaluation.TripletClassificationEvaluator method)
AnalogyModel (class in torchkge.models.bilinear)
B
bern_probs (torchkge.sampling.BernoulliNegativeSampler attribute)
BernoulliNegativeSampler (class in torchkge.sampling)
BinaryCrossEntropyLoss (class in torchkge.utils.losses)
C
cartesian_product_relations() (in module torchkge.utils.data_redundancy)
complex_dim (torchkge.models.bilinear.AnalogyModel attribute)
ComplExModel (class in torchkge.models.bilinear)
ConvKBModel (class in torchkge.models.deep)
corrupt_batch() (torchkge.sampling.BernoulliNegativeSampler method)
(torchkge.sampling.PositionalNegativeSampler method)
(torchkge.sampling.UniformNegativeSampler method)
count_triplets() (in module torchkge.utils.data_redundancy)
D
dictionary (torchkge.inference.EntityInference attribute)
(torchkge.inference.RelationInference attribute)
directed (torchkge.evaluation.RelationPredictionEvaluator attribute)
dissimilarity (torchkge.models.interfaces.TranslationModel attribute)
DistMultModel (class in torchkge.models.bilinear)
duplicates() (in module torchkge.utils.data_redundancy)
E
el2_torus_dissimilarity() (in module torchkge.utils.dissimilarities)
emb_dim (torchkge.models.translation.TorusEModel attribute)
(torchkge.models.translation.TransEModel attribute)
(torchkge.models.translation.TransHModel attribute)
ent2ix (torchkge.data_structures.KnowledgeGraph attribute)
ent_emb (torchkge.models.bilinear.DistMultModel attribute)
(torchkge.models.bilinear.HolEModel attribute)
(torchkge.models.bilinear.RESCALModel attribute)
(torchkge.models.deep.ConvKBModel attribute)
(torchkge.models.translation.TorusEModel attribute)
(torchkge.models.translation.TransDModel attribute)
(torchkge.models.translation.TransEModel attribute)
(torchkge.models.translation.TransHModel attribute)
(torchkge.models.translation.TransRModel attribute)
ent_emb_dim (torchkge.models.translation.TransDModel attribute)
(torchkge.models.translation.TransRModel attribute)
ent_proj_vect (torchkge.models.translation.TransDModel attribute)
entities1 (torchkge.inference.RelationInference attribute)
entities2 (torchkge.inference.RelationInference attribute)
EntityInference (class in torchkge.inference)
evaluate() (torchkge.evaluation.LinkPredictionEvaluator method)
(torchkge.evaluation.TripletClassificationEvaluator method)
evaluate_dicts() (torchkge.data_structures.KnowledgeGraph method)
evaluate_probabilities() (torchkge.sampling.BernoulliNegativeSampler method)
evaluate_projections() (torchkge.models.translation.TransHModel method)
evaluate_projectionss() (torchkge.models.translation.TransDModel method)
(torchkge.models.translation.TransRModel method)
evaluated (torchkge.evaluation.LinkPredictionEvaluator attribute)
(torchkge.evaluation.RelationPredictionEvaluator attribute)
(torchkge.evaluation.TripletClassificationEvaluator attribute)
evaluated_projections (torchkge.models.translation.TransDModel attribute)
(torchkge.models.translation.TransRModel attribute)
F
filt_rank_true_heads (torchkge.evaluation.LinkPredictionEvaluator attribute)
filt_rank_true_rels (torchkge.evaluation.RelationPredictionEvaluator attribute)
filt_rank_true_tails (torchkge.evaluation.LinkPredictionEvaluator attribute)
find_possibilities() (torchkge.sampling.PositionalNegativeSampler method)
forward() (torchkge.models.interfaces.Model method)
(torchkge.utils.losses.BinaryCrossEntropyLoss method)
(torchkge.utils.losses.LogisticLoss method)
(torchkge.utils.losses.MarginLoss method)
G
get_counter_examples() (torchkge.utils.training.Trainer method)
get_df() (torchkge.data_structures.KnowledgeGraph method)
get_embeddings() (torchkge.models.bilinear.AnalogyModel method)
(torchkge.models.bilinear.ComplExModel method)
(torchkge.models.bilinear.DistMultModel method)
(torchkge.models.bilinear.HolEModel method)
(torchkge.models.bilinear.RESCALModel method)
(torchkge.models.deep.ConvKBModel method)
(torchkge.models.interfaces.Model method)
(torchkge.models.interfaces.TranslationModel method)
(torchkge.models.translation.TorusEModel method)
(torchkge.models.translation.TransDModel method)
(torchkge.models.translation.TransEModel method)
(torchkge.models.translation.TransHModel method)
(torchkge.models.translation.TransRModel method)
get_mask() (torchkge.data_structures.KnowledgeGraph method)
get_rolling_matrix() (torchkge.models.bilinear.HolEModel static method)
get_scores() (torchkge.evaluation.TripletClassificationEvaluator method)
get_sizes() (torchkge.data_structures.KnowledgeGraph static method)
H
head_idx (torchkge.data_structures.KnowledgeGraph attribute)
hit_at_k() (torchkge.evaluation.LinkPredictionEvaluator method)
HolEModel (class in torchkge.models.bilinear)
I
im_ent_emb (torchkge.models.bilinear.AnalogyModel attribute)
(torchkge.models.bilinear.ComplExModel attribute)
im_rel_emb (torchkge.models.bilinear.AnalogyModel attribute)
(torchkge.models.bilinear.ComplExModel attribute)
inference_prepare_candidates() (torchkge.models.bilinear.AnalogyModel method)
(torchkge.models.bilinear.ComplExModel method)
(torchkge.models.bilinear.DistMultModel method)
(torchkge.models.bilinear.HolEModel method)
(torchkge.models.bilinear.RESCALModel method)
(torchkge.models.deep.ConvKBModel method)
(torchkge.models.interfaces.Model method)
(torchkge.models.interfaces.TranslationModel method)
(torchkge.models.translation.TorusEModel method)
(torchkge.models.translation.TransDModel method)
(torchkge.models.translation.TransEModel method)
(torchkge.models.translation.TransHModel method)
(torchkge.models.translation.TransRModel method)
inference_scoring_function() (torchkge.models.bilinear.AnalogyModel method)
(torchkge.models.bilinear.ComplExModel method)
(torchkge.models.bilinear.DistMultModel method)
(torchkge.models.bilinear.HolEModel method)
(torchkge.models.bilinear.RESCALModel method)
(torchkge.models.deep.ConvKBModel method)
(torchkge.models.interfaces.Model method)
(torchkge.models.interfaces.TranslationModel method)
K
kg (torchkge.evaluation.LinkPredictionEvaluator attribute)
(torchkge.evaluation.RelationPredictionEvaluator attribute)
kg_test (torchkge.evaluation.TripletClassificationEvaluator attribute)
kg_val (torchkge.evaluation.TripletClassificationEvaluator attribute)
KnowledgeGraph (class in torchkge.data_structures)
known_entities (torchkge.inference.EntityInference attribute)
known_relations (torchkge.inference.EntityInference attribute)
L
l1_dissimilarity() (in module torchkge.utils.dissimilarities)
l1_torus_dissimilarity() (in module torchkge.utils.dissimilarities)
l2_dissimilarity() (in module torchkge.utils.dissimilarities)
l2_torus_dissimilarity() (in module torchkge.utils.dissimilarities)
LinkPredictionEvaluator (class in torchkge.evaluation)
load_fb13() (in module torchkge.utils.datasets)
load_fb15k() (in module torchkge.utils.datasets)
load_fb15k237() (in module torchkge.utils.datasets)
load_pretrained_complex() (in module torchkge.utils.pretrained_models)
load_pretrained_rescal() (in module torchkge.utils.pretrained_models)
load_pretrained_transe() (in module torchkge.utils.pretrained_models)
load_wikidata_vitals() (in module torchkge.utils.datasets)
load_wikidatasets() (in module torchkge.utils.datasets)
load_wn18() (in module torchkge.utils.datasets)
load_wn18rr() (in module torchkge.utils.datasets)
load_yago3_10() (in module torchkge.utils.datasets)
LogisticLoss (class in torchkge.utils.losses)
M
MarginLoss (class in torchkge.utils.losses)
mean_rank() (torchkge.evaluation.LinkPredictionEvaluator method)
missing (torchkge.inference.EntityInference attribute)
Model (class in torchkge.models.interfaces)
model (torchkge.evaluation.LinkPredictionEvaluator attribute)
(torchkge.evaluation.RelationPredictionEvaluator attribute)
(torchkge.evaluation.TripletClassificationEvaluator attribute)
(torchkge.inference.EntityInference attribute)
(torchkge.inference.RelationInference attribute)
mrr() (torchkge.evaluation.LinkPredictionEvaluator method)
N
n_ent (torchkge.data_structures.KnowledgeGraph attribute)
(torchkge.models.interfaces.Model attribute)
n_facts (torchkge.data_structures.KnowledgeGraph attribute)
n_poss_heads (torchkge.sampling.PositionalNegativeSampler attribute)
n_poss_tails (torchkge.sampling.PositionalNegativeSampler attribute)
n_rel (torchkge.data_structures.KnowledgeGraph attribute)
(torchkge.models.interfaces.Model attribute)
norm_vect (torchkge.models.translation.TransHModel attribute)
normalize_parameters() (torchkge.models.bilinear.AnalogyModel method)
(torchkge.models.bilinear.ComplExModel method)
(torchkge.models.bilinear.DistMultModel method)
(torchkge.models.bilinear.HolEModel method)
(torchkge.models.bilinear.RESCALModel method)
(torchkge.models.deep.ConvKBModel method)
(torchkge.models.interfaces.Model method)
(torchkge.models.interfaces.TranslationModel method)
(torchkge.models.translation.TorusEModel method)
(torchkge.models.translation.TransDModel method)
(torchkge.models.translation.TransEModel method)
(torchkge.models.translation.TransHModel method)
(torchkge.models.translation.TransRModel method)
P
PositionalNegativeSampler (class in torchkge.sampling)
possible_heads (torchkge.sampling.PositionalNegativeSampler attribute)
possible_tails (torchkge.sampling.PositionalNegativeSampler attribute)
predictions (torchkge.inference.EntityInference attribute)
(torchkge.inference.RelationInference attribute)
print_results() (torchkge.evaluation.LinkPredictionEvaluator method)
proj_mat (torchkge.models.translation.TransRModel attribute)
project() (torchkge.models.translation.TransDModel method)
projected_entities (torchkge.models.translation.TransDModel attribute)
(torchkge.models.translation.TransRModel attribute)
R
rank_true_heads (torchkge.evaluation.LinkPredictionEvaluator attribute)
rank_true_rels (torchkge.evaluation.RelationPredictionEvaluator attribute)
rank_true_tails (torchkge.evaluation.LinkPredictionEvaluator attribute)
re_ent_emb (torchkge.models.bilinear.AnalogyModel attribute)
(torchkge.models.bilinear.ComplExModel attribute)
re_rel_emb (torchkge.models.bilinear.AnalogyModel attribute)
(torchkge.models.bilinear.ComplExModel attribute)
rel2ix (torchkge.data_structures.KnowledgeGraph attribute)
rel_emb (torchkge.models.bilinear.DistMultModel attribute)
(torchkge.models.bilinear.HolEModel attribute)
(torchkge.models.deep.ConvKBModel attribute)
(torchkge.models.translation.TorusEModel attribute)
(torchkge.models.translation.TransDModel attribute)
(torchkge.models.translation.TransEModel attribute)
(torchkge.models.translation.TransHModel attribute)
(torchkge.models.translation.TransRModel attribute)
rel_emb_dim (torchkge.models.translation.TransDModel attribute)
(torchkge.models.translation.TransRModel attribute)
rel_mat (torchkge.models.bilinear.RESCALModel attribute)
rel_proj_vect (torchkge.models.translation.TransDModel attribute)
RelationInference (class in torchkge.inference)
RelationPredictionEvaluator (class in torchkge.evaluation)
relations (torchkge.data_structures.KnowledgeGraph attribute)
RESCALModel (class in torchkge.models.bilinear)
S
sampler (torchkge.evaluation.TripletClassificationEvaluator attribute)
sc_ent_emb (torchkge.models.bilinear.AnalogyModel attribute)
sc_rel_emb (torchkge.models.bilinear.AnalogyModel attribute)
scalar_dim (torchkge.models.bilinear.AnalogyModel attribute)
scores (torchkge.inference.EntityInference attribute)
(torchkge.inference.RelationInference attribute)
scoring_function() (torchkge.models.bilinear.AnalogyModel method)
(torchkge.models.bilinear.ComplExModel method)
(torchkge.models.bilinear.DistMultModel method)
(torchkge.models.bilinear.HolEModel method)
(torchkge.models.bilinear.RESCALModel method)
(torchkge.models.deep.ConvKBModel method)
(torchkge.models.interfaces.Model method)
(torchkge.models.interfaces.TranslationModel method)
(torchkge.models.translation.TorusEModel method)
(torchkge.models.translation.TransDModel method)
(torchkge.models.translation.TransEModel method)
(torchkge.models.translation.TransHModel method)
(torchkge.models.translation.TransRModel method)
SmallKG (class in torchkge.data_structures)
split_kg() (torchkge.data_structures.KnowledgeGraph method)
T
tail_idx (torchkge.data_structures.KnowledgeGraph attribute)
thresholds (torchkge.evaluation.TripletClassificationEvaluator attribute)
top_k (torchkge.inference.EntityInference attribute)
(torchkge.inference.RelationInference attribute)
TorusEModel (class in torchkge.models.translation)
TrainDataLoader (class in torchkge.utils.training)
Trainer (class in torchkge.utils.training)
TransDModel (class in torchkge.models.translation)
TransEModel (class in torchkge.models.translation)
TransHModel (class in torchkge.models.translation)
TranslationModel (class in torchkge.models.interfaces)
TransRModel (class in torchkge.models.translation)
TripletClassificationEvaluator (class in torchkge.evaluation)
U
UniformNegativeSampler (class in torchkge.sampling)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds