Index

A

AllScoresBESS (class in besskge.bess)

AllScoresPipeline (class in besskge.pipeline)

B

BaseLossFunction (class in besskge.loss) BaseMetric (class in besskge.metric) BaseScoreFunction (class in besskge.scoring) besskge module BessKGE (class in besskge.bess) besskge.batch_sampler module besskge.bess module besskge.dataset module besskge.embedding module besskge.loss module besskge.metric module besskge.negative_sampler module besskge.pipeline module besskge.scoring module

besskge.sharding module besskge.utils module BoxE (class in besskge.scoring) boxe_score() (besskge.scoring.BoxE method) broadcasted_distance() (besskge.scoring.BoxE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.InterHT method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method) broadcasted_dot_product() (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistMult method) (besskge.scoring.MatrixDecompositionScoreFunction method) build_ogbl_biokg() (besskge.dataset.KGDataset class method) build_ogbl_wikikg2() (besskge.dataset.KGDataset class method) build_openbiolink() (besskge.dataset.KGDataset class method) build_yago310() (besskge.dataset.KGDataset class method)

C

ComplEx (class in besskge.scoring) complex_multiplication() (in module besskge.utils) complex_rotation() (in module besskge.utils) ConvE (class in besskge.scoring) corruption_scheme (besskge.negative_sampler.PlaceholderNegativeSampler attribute) (besskge.negative_sampler.RandomShardedNegativeSampler attribute) (besskge.negative_sampler.ShardedNegativeSampler attribute) (besskge.negative_sampler.TripleBasedShardedNegativeSampler attribute) (besskge.negative_sampler.TypeBasedShardedNegativeSampler attribute)

create() (besskge.sharding.Sharding class method) create_from_dataset() (besskge.sharding.PartitionedTripleSet class method) create_from_queries() (besskge.sharding.PartitionedTripleSet class method)

D

dict_metrics_from_ranks() (besskge.metric.Evaluation method) DistanceBasedScoreFunction (class in besskge.scoring)

DistMult (class in besskge.scoring) dummy (besskge.sharding.PartitionedTripleSet attribute)

E

EmbeddingMovingBessKGE (class in besskge.bess) entity_dict (besskge.dataset.KGDataset attribute) entity_embedding (besskge.scoring.BaseScoreFunction attribute) (besskge.scoring.BoxE attribute) (besskge.scoring.ComplEx attribute) (besskge.scoring.ConvE attribute) (besskge.scoring.DistanceBasedScoreFunction attribute) (besskge.scoring.DistMult attribute) (besskge.scoring.InterHT attribute) (besskge.scoring.MatrixDecompositionScoreFunction attribute) (besskge.scoring.PairRE attribute) (besskge.scoring.RotatE attribute) (besskge.scoring.TranS attribute) (besskge.scoring.TransE attribute) (besskge.scoring.TripleRE attribute)

entity_to_idx (besskge.sharding.Sharding attribute) entity_to_shard (besskge.sharding.Sharding attribute) entity_type_counts (besskge.sharding.Sharding attribute) entity_type_offsets (besskge.sharding.Sharding attribute) Evaluation (class in besskge.metric)

F

flat_negative_format (besskge.negative_sampler.PlaceholderNegativeSampler attribute) (besskge.negative_sampler.RandomShardedNegativeSampler attribute) (besskge.negative_sampler.ShardedNegativeSampler attribute) (besskge.negative_sampler.TripleBasedShardedNegativeSampler attribute) (besskge.negative_sampler.TypeBasedShardedNegativeSampler attribute) forward() (besskge.bess.AllScoresBESS method) (besskge.bess.BessKGE method) (besskge.bess.EmbeddingMovingBessKGE method) (besskge.bess.ScoreMovingBessKGE method) (besskge.bess.TopKQueryBessKGE method) (besskge.loss.BaseLossFunction method) (besskge.loss.LogSigmoidLoss method) (besskge.loss.MarginBasedLossFunction method) (besskge.loss.MarginRankingLoss method) (besskge.loss.SampledSoftmaxCrossEntropyLoss method) (besskge.pipeline.AllScoresPipeline method) (besskge.scoring.BaseScoreFunction method) (besskge.scoring.BoxE method) (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.DistMult method) (besskge.scoring.InterHT method) (besskge.scoring.MatrixDecompositionScoreFunction method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method)

from_dataframe() (besskge.dataset.KGDataset class method) from_triples() (besskge.dataset.KGDataset class method)

G

gather_indices() (in module besskge.utils) get_dataloader() (besskge.batch_sampler.RandomShardedBatchSampler method) (besskge.batch_sampler.RigidShardedBatchSampler method) (besskge.batch_sampler.ShardedBatchSampler method) get_dataloader_sampler() (besskge.batch_sampler.RandomShardedBatchSampler method) (besskge.batch_sampler.RigidShardedBatchSampler method) (besskge.batch_sampler.ShardedBatchSampler method)

get_entity_filter() (in module besskge.utils) get_negative_weights() (besskge.loss.BaseLossFunction method) (besskge.loss.LogSigmoidLoss method) (besskge.loss.MarginBasedLossFunction method) (besskge.loss.MarginRankingLoss method) (besskge.loss.SampledSoftmaxCrossEntropyLoss method)

H

HitsAtK (class in besskge.metric)

ht_types (besskge.dataset.KGDataset property)

I

init_KGE_normal() (in module besskge.embedding) init_KGE_uniform() (in module besskge.embedding) init_uniform_norm() (in module besskge.embedding) init_uniform_rotation() (in module besskge.embedding)

init_xavier_norm() (in module besskge.embedding) initialize_entity_embedding() (in module besskge.embedding) initialize_relation_embedding() (in module besskge.embedding) InterHT (class in besskge.scoring) inverse_triples (besskge.sharding.PartitionedTripleSet attribute)

K

KGDataset (class in besskge.dataset)

L

load() (besskge.dataset.KGDataset class method) (besskge.sharding.Sharding class method) local_sampling (besskge.negative_sampler.PlaceholderNegativeSampler attribute) (besskge.negative_sampler.RandomShardedNegativeSampler attribute) (besskge.negative_sampler.ShardedNegativeSampler attribute) (besskge.negative_sampler.TripleBasedShardedNegativeSampler attribute) (besskge.negative_sampler.TypeBasedShardedNegativeSampler attribute)

LogSigmoidLoss (class in besskge.loss) loss_scale (besskge.loss.BaseLossFunction attribute) (besskge.loss.LogSigmoidLoss attribute) (besskge.loss.MarginBasedLossFunction attribute) (besskge.loss.MarginRankingLoss attribute) (besskge.loss.SampledSoftmaxCrossEntropyLoss attribute)

M

MarginBasedLossFunction (class in besskge.loss) MarginRankingLoss (class in besskge.loss) MatrixDecompositionScoreFunction (class in besskge.scoring) max_entity_per_shard (besskge.sharding.Sharding property) METRICS_DICT (in module besskge.metric) module besskge besskge.batch_sampler besskge.bess besskge.dataset besskge.embedding besskge.loss besskge.metric besskge.negative_sampler besskge.pipeline besskge.scoring besskge.sharding besskge.utils

N

n_embedding_parameters (besskge.bess.BessKGE property) (besskge.bess.EmbeddingMovingBessKGE property) (besskge.bess.ScoreMovingBessKGE property) n_entity (besskge.dataset.KGDataset attribute) (besskge.sharding.Sharding property) n_relation_type (besskge.dataset.KGDataset attribute) n_shard (besskge.sharding.Sharding attribute) neg_heads (besskge.dataset.KGDataset attribute) (besskge.sharding.PartitionedTripleSet attribute) neg_tails (besskge.dataset.KGDataset attribute) (besskge.sharding.PartitionedTripleSet attribute) negative_adversarial_sampling (besskge.loss.BaseLossFunction attribute) (besskge.loss.LogSigmoidLoss attribute) (besskge.loss.MarginBasedLossFunction attribute) (besskge.loss.MarginRankingLoss attribute) (besskge.loss.SampledSoftmaxCrossEntropyLoss attribute) negative_adversarial_scale (besskge.loss.BaseLossFunction attribute) (besskge.loss.LogSigmoidLoss attribute) (besskge.loss.MarginBasedLossFunction attribute) (besskge.loss.MarginRankingLoss attribute) (besskge.loss.SampledSoftmaxCrossEntropyLoss attribute)

negative_sample_sharing (besskge.scoring.BaseScoreFunction attribute) (besskge.scoring.BoxE attribute) (besskge.scoring.ComplEx attribute) (besskge.scoring.ConvE attribute) (besskge.scoring.DistanceBasedScoreFunction attribute) (besskge.scoring.DistMult attribute) (besskge.scoring.InterHT attribute) (besskge.scoring.MatrixDecompositionScoreFunction attribute) (besskge.scoring.PairRE attribute) (besskge.scoring.RotatE attribute) (besskge.scoring.TranS attribute) (besskge.scoring.TransE attribute) (besskge.scoring.TripleRE attribute)

O

original_triple_ids (besskge.dataset.KGDataset attribute)

P

pad_negatives() (besskge.negative_sampler.TripleBasedShardedNegativeSampler method) PairRE (class in besskge.scoring)

partition_mode (besskge.sharding.PartitionedTripleSet attribute) PartitionedTripleSet (class in besskge.sharding) PlaceholderNegativeSampler (class in besskge.negative_sampler)

R

RandomShardedBatchSampler (class in besskge.batch_sampler) RandomShardedNegativeSampler (class in besskge.negative_sampler) ranks_from_indices() (besskge.metric.Evaluation method) ranks_from_scores() (besskge.metric.Evaluation method) ReciprocalRank (class in besskge.metric) reduce_embedding() (besskge.scoring.BoxE method) (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.DistMult method) (besskge.scoring.InterHT method) (besskge.scoring.MatrixDecompositionScoreFunction method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method) refactor_embedding_sharding() (in module besskge.embedding) relation_dict (besskge.dataset.KGDataset attribute)

relation_embedding (besskge.scoring.BaseScoreFunction attribute) (besskge.scoring.BoxE attribute) (besskge.scoring.ComplEx attribute) (besskge.scoring.ConvE attribute) (besskge.scoring.DistanceBasedScoreFunction attribute) (besskge.scoring.DistMult attribute) (besskge.scoring.InterHT attribute) (besskge.scoring.MatrixDecompositionScoreFunction attribute) (besskge.scoring.PairRE attribute) (besskge.scoring.RotatE attribute) (besskge.scoring.TranS attribute) (besskge.scoring.TransE attribute) (besskge.scoring.TripleRE attribute) RigidShardedBatchSampler (class in besskge.batch_sampler) rng (besskge.negative_sampler.PlaceholderNegativeSampler attribute) (besskge.negative_sampler.RandomShardedNegativeSampler attribute) (besskge.negative_sampler.ShardedNegativeSampler attribute) (besskge.negative_sampler.TripleBasedShardedNegativeSampler attribute) (besskge.negative_sampler.TypeBasedShardedNegativeSampler attribute) RotatE (class in besskge.scoring)

S

sample_triples() (besskge.batch_sampler.RandomShardedBatchSampler method) (besskge.batch_sampler.RigidShardedBatchSampler method) (besskge.batch_sampler.ShardedBatchSampler method) SampledSoftmaxCrossEntropyLoss (class in besskge.loss) save() (besskge.dataset.KGDataset method) (besskge.sharding.Sharding method) score_batch() (besskge.bess.BessKGE method) (besskge.bess.EmbeddingMovingBessKGE method) (besskge.bess.ScoreMovingBessKGE method) score_heads() (besskge.scoring.BaseScoreFunction method) (besskge.scoring.BoxE method) (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.DistMult method) (besskge.scoring.InterHT method) (besskge.scoring.MatrixDecompositionScoreFunction method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method) score_tails() (besskge.scoring.BaseScoreFunction method) (besskge.scoring.BoxE method) (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.DistMult method) (besskge.scoring.InterHT method) (besskge.scoring.MatrixDecompositionScoreFunction method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method)

score_triple() (besskge.scoring.BaseScoreFunction method) (besskge.scoring.BoxE method) (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.DistMult method) (besskge.scoring.InterHT method) (besskge.scoring.MatrixDecompositionScoreFunction method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method) ScoreMovingBessKGE (class in besskge.bess) shard_and_idx_to_entity (besskge.sharding.Sharding attribute) shard_counts (besskge.sharding.Sharding attribute) shard_negatives() (besskge.negative_sampler.TripleBasedShardedNegativeSampler method) ShardedBatchSampler (class in besskge.batch_sampler) ShardedNegativeSampler (class in besskge.negative_sampler) sharding (besskge.scoring.BaseScoreFunction attribute) (besskge.scoring.BoxE attribute) (besskge.scoring.ComplEx attribute) (besskge.scoring.ConvE attribute) (besskge.scoring.DistanceBasedScoreFunction attribute) (besskge.scoring.DistMult attribute) (besskge.scoring.InterHT attribute) (besskge.scoring.MatrixDecompositionScoreFunction attribute) (besskge.scoring.PairRE attribute) (besskge.scoring.RotatE attribute) (besskge.scoring.TranS attribute) (besskge.scoring.TransE attribute) (besskge.scoring.TripleRE attribute) (besskge.sharding.PartitionedTripleSet attribute) Sharding (class in besskge.sharding) stacked_metrics_from_ranks() (besskge.metric.Evaluation method)

T

TopKQueryBessKGE (class in besskge.bess) TranS (class in besskge.scoring) TransE (class in besskge.scoring) triple_counts (besskge.sharding.PartitionedTripleSet attribute) triple_offsets (besskge.sharding.PartitionedTripleSet attribute) triple_sort_idx (besskge.sharding.PartitionedTripleSet attribute)

TripleBasedShardedNegativeSampler (class in besskge.negative_sampler) TripleRE (class in besskge.scoring) triples (besskge.dataset.KGDataset attribute) (besskge.sharding.PartitionedTripleSet attribute) type_offsets (besskge.dataset.KGDataset attribute) TypeBasedShardedNegativeSampler (class in besskge.negative_sampler) types (besskge.sharding.PartitionedTripleSet attribute)

U

update_sharding() (besskge.scoring.BaseScoreFunction method) (besskge.scoring.BoxE method) (besskge.scoring.ComplEx method) (besskge.scoring.ConvE method) (besskge.scoring.DistanceBasedScoreFunction method) (besskge.scoring.DistMult method) (besskge.scoring.InterHT method) (besskge.scoring.MatrixDecompositionScoreFunction method) (besskge.scoring.PairRE method) (besskge.scoring.RotatE method) (besskge.scoring.TranS method) (besskge.scoring.TransE method) (besskge.scoring.TripleRE method)

W

worker_init_fn() (besskge.batch_sampler.RandomShardedBatchSampler static method) (besskge.batch_sampler.RigidShardedBatchSampler static method) (besskge.batch_sampler.ShardedBatchSampler static method)