pykeen
Getting Started
Installation
First Steps
Knowledge Graph Embedding Models
Tracking Results during Training
Saving Checkpoints during Training
A Toy Example with Translational Distance Models
Understanding the Evaluation
Optimizing a Model’s Hyper-parameters
Running an Ablation Study
Performance Tricks
Representations
Getting Started with NodePiece
Inductive Link Prediction
Splitting
PyTorch Lightning Integration
Using Resolvers
Troubleshooting
Bring Your Own
Bring Your Own Data
Bring Your Own Interaction
Extending PyKEEN
Extending the Datasets
Extending the Models
Reference
Pipeline
Models
Datasets
Inductive Datasets
Entity Alignment
Triples
Triples Workflows
Training
Stoppers
Loss Functions
Regularizers
Result Trackers
Negative Sampling
Filtering
Optimizers
Evaluation
Metrics
Hyper-parameter Optimization
Ablation
Prediction
Uncertainty
Sealant
Constants
Flexible Weight Checkpoints
pykeen.nn
Utilities
Appendix
Analysis
References
pykeen
Index
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
|
Z
_
__call__() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(AllScoreConsumer method)
(AnchorSearcher method)
(AnchorSelection method)
(AnchorTokenizer method)
(ArithmeticMeanRank method)
(BestCheckpointKeeper method)
(BestCheckpointSchedule method)
(CheckpointKeeper method)
(CheckpointSchedule method)
(ClassificationMetric method)
(Cleaner method)
(Count method)
(CountScoreConsumer method)
(CSGraphAnchorSearcher method)
(DerivedRankBasedMetric method)
(EveryCheckpointSchedule method)
(ExpectationNormalizedMetric method)
(ExplicitCheckpointKeeper method)
(ExplicitCheckpointSchedule method)
(GalkinPrecomputedTokenizerLoader method)
(GeometricMeanRank method)
(GraphPairCombinator method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(LastCheckpointKeeper method)
(MedianRank method)
(MetisAnchorTokenizer method)
(MixtureAnchorSelection method)
(ModuloCheckpointKeeper method)
(PersonalizedPageRankAnchorSearcher method)
(PrecomputedPoolTokenizer method)
(PrecomputedTokenizerLoader method)
(PretrainedInitializer method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(RelationTokenizer method)
(ScipySparseAnchorSearcher method)
(ScoreConsumer method)
(SingleSelection method)
(SparseBFSSearcher method)
(StandardDeviation method)
(Tokenizer method)
(TopKScoreConsumer method)
(TorchPrecomputedTokenizerLoader method)
(UnionCheckpointKeeper method)
(UnionCheckpointSchedule method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
A
ablation_pipeline() (in module pykeen.ablation)
add() (BloomFilterer method)
add_entity_labels() (in module pykeen.triples.analysis)
add_membership_columns() (Predictions method)
add_parameter() (Regularizer method)
add_relation_labels() (in module pykeen.triples.analysis)
adjacency_tensor_to_stacked_matrix() (in module pykeen.nn.utils)
adjust() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(ReindexedMetric method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
AdjustedArithmeticMeanRank (class in pykeen.metrics.ranking)
AdjustedArithmeticMeanRankIndex (class in pykeen.metrics.ranking)
AdjustedGeometricMeanRankIndex (class in pykeen.metrics.ranking)
AdjustedHitsAtK (class in pykeen.metrics.ranking)
AdjustedInverseHarmonicMeanRank (class in pykeen.metrics.ranking)
AdversarialBCEWithLogitsLoss (class in pykeen.losses)
AdversarialLoss (class in pykeen.losses)
AffineTransformationParameters (class in pykeen.metrics.ranking)
alignment (ProcessedTuple attribute)
ALIGNMENT_RELATION_NAME (ExtraRelationGraphPairCombinator attribute)
all_in_bounds() (in module pykeen.utils)
AllPredictionDataset (class in pykeen.predict)
AllScoreConsumer (class in pykeen.predict)
AnchorSearcher (class in pykeen.nn.node_piece)
AnchorSelection (class in pykeen.nn.node_piece)
AnchorTokenizer (class in pykeen.nn.node_piece)
AnyTriples (in module pykeen.triples)
append_weight_regularizer() (ERModel method)
apply() (Sealant method)
apply_only_once (Regularizer attribute)
apply_optional_bn() (in module pykeen.nn.utils)
approximate() (LowRankRepresentation class method)
(ValueRange method)
AristoV4 (class in pykeen.datasets)
ArithmeticMeanRank (class in pykeen.metrics.ranking)
as_embedding() (PretrainedInitializer method)
at_least_eps() (in module pykeen.utils)
AttentionEdgeWeighting (class in pykeen.nn.weighting)
AutoSF (class in pykeen.models)
AutoSFInteraction (class in pykeen.nn.modules)
B
BackfillRepresentation (class in pykeen.nn.representation)
base_cls (DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(ReindexedMetric attribute)
(ZMetric attribute)
base_file_name (CoreTriplesFactory attribute)
base_weights (BasesDecomposition attribute)
bases (BasesDecomposition attribute)
bases_kwargs (UnionCheckpointKeeper attribute)
(UnionCheckpointSchedule attribute)
BasesDecomposition (class in pykeen.nn.message_passing)
BasicNegativeSampler (class in pykeen.sampling)
batch_size_search() (TrainingLoop method)
batched_dot() (in module pykeen.utils)
BCEAfterSigmoidLoss (class in pykeen.losses)
BCEWithLogitsLoss (class in pykeen.losses)
BernoulliNegativeSampler (class in pykeen.sampling)
best_epoch (EarlyStopper attribute)
best_metric (EarlyStopper attribute)
best_model_path (EarlyStopper attribute)
BestCheckpointKeeper (class in pykeen.checkpoints)
BestCheckpointSchedule (class in pykeen.checkpoints)
bfs() (ScipySparseAnchorSearcher static method)
(SparseBFSSearcher static method)
Bias (class in pykeen.utils)
binarize (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(ClassificationMetric attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(Metric attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
BioKG (class in pykeen.datasets)
BiomedicalCURIERepresentation (class in pykeen.nn.representation)
BlockDecomposition (class in pykeen.nn.message_passing)
BloomFilterer (class in pykeen.sampling.filtering)
BoxE (class in pykeen.models)
boxe_kg_arity_position_score() (BoxEInteraction class method)
BoxEInteraction (class in pykeen.nn.modules)
broadcast_upgrade_to_sequences() (in module pykeen.utils)
C
CachedTextRepresentation (class in pykeen.nn.representation)
calculate_broadcasted_elementwise_result_shape() (in module pykeen.utils)
callback_resolver (in module pykeen.training.callbacks)
cast_constrainer() (in module pykeen.typing)
CharacterEmbeddingTextEncoder (class in pykeen.nn.text)
check_assignment() (TensorTrainRepresentation static method)
check_factors() (TensorTrainRepresentation static method)
check_shapes() (in module pykeen.utils)
CheckpointKeeper (class in pykeen.checkpoints)
CheckpointSchedule (class in pykeen.checkpoints)
CheckpointTrainingCallback (class in pykeen.training.callbacks)
checksum (TrainingLoop attribute)
CircularCorrelationCompositionModule (class in pykeen.nn.compositions)
CKG (class in pykeen.datasets)
clamp_norm() (in module pykeen.utils)
ClampedInteraction (class in pykeen.nn.modules)
ClassificationEvaluator (class in pykeen.evaluation)
ClassificationMetric (class in pykeen.metrics)
ClassificationMetricResults (class in pykeen.evaluation)
clean_up_checkpoint (EarlyStopper attribute)
Cleaner (class in pykeen.triples.splitting)
cleanup_pair() (Cleaner method)
(DeterministicCleaner method)
(RandomizedCleaner method)
CleanupSplitter (class in pykeen.triples.splitting)
clear() (ClassificationEvaluator method)
(Evaluator method)
(MacroRankBasedEvaluator method)
(RankBasedEvaluator method)
cli() (Dataset class method)
,
[1]
clone_and_exchange_triples() (CoreTriplesFactory method)
(TriplesFactory method)
(TriplesNumericLiteralsFactory method)
closed_expectation (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(ClassificationMetric attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(Metric attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
closed_variance (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(ClassificationMetric attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(Metric attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
CN3l (class in pykeen.datasets)
CoDExLarge (class in pykeen.datasets)
CoDExMedium (class in pykeen.datasets)
CoDExSmall (class in pykeen.datasets)
CollapseGraphPairCombinator (class in pykeen.datasets.ea.combination)
collate() (SLCWAInstances static method)
collect_regularization_term() (CooccurrenceFilteredModel method)
(EvaluationOnlyModel method)
(FixedModel method)
(Model method)
Combination (class in pykeen.nn.combination)
combine() (CombinedRepresentation static method)
combine_complex() (in module pykeen.utils)
CombinedCompGCNRepresentations (class in pykeen.nn.representation)
CombinedRegularizer (class in pykeen.regularizers)
CombinedRepresentation (class in pykeen.nn.representation)
compact_mapping() (in module pykeen.utils)
CompGCN (class in pykeen.models)
CompGCNLayer (class in pykeen.nn.representation)
ComplEx (class in pykeen.models)
complex_normalize() (in module pykeen.utils)
ComplExInteraction (class in pykeen.nn.modules)
ComplExLiteral (class in pykeen.models)
ComplexSeparatedCombination (class in pykeen.nn.combination)
compose (class in pykeen.utils)
CompositionModule (class in pykeen.nn.compositions)
CompressedSingleDataset (class in pykeen.datasets.base)
compute_box() (BoxEInteraction class method)
compute_compressed_adjacency_list() (in module pykeen.triples.utils)
ConcatAggregationCombination (class in pykeen.nn.combination)
ConcatCombination (class in pykeen.nn.combination)
ConcatMLP (class in pykeen.nn.perceptron)
ConcatProjectionCombination (class in pykeen.nn.combination)
ConceptNet (class in pykeen.datasets)
configure_optimizers() (LitModule method)
ConsoleResultTracker (class in pykeen.trackers)
Constrainer (in module pykeen.typing)
constrainer_default_kwargs (KG2E attribute)
consume_scores() (in module pykeen.predict)
contains() (BloomFilterer method)
(Filterer method)
(PythonSetFilterer method)
ConvE (class in pykeen.models)
ConvEInteraction (class in pykeen.nn.modules)
ConvKB (class in pykeen.models)
ConvKBInteraction (class in pykeen.nn.modules)
CooccurrenceFilteredModel (class in pykeen.models)
CoreTriplesFactory (class in pykeen.triples)
corrupt_batch() (BasicNegativeSampler method)
(BernoulliNegativeSampler method)
(NegativeSampler method)
(PseudoTypedNegativeSampler method)
Count (class in pykeen.metrics.ranking)
count() (AffineTransformationParameters method)
Countries (class in pykeen.datasets)
CountScoreConsumer (class in pykeen.predict)
CoverageSplitter (class in pykeen.triples.splitting)
CP (class in pykeen.models)
CPInteraction (class in pykeen.nn.modules)
create() (CoreTriplesFactory class method)
create_adjacency() (ScipySparseAnchorSearcher static method)
(SparseBFSSearcher static method)
create_default_assignment() (TensorTrainRepresentation static method)
create_entity_representation_for_new_triples() (InductiveNodePiece method)
create_inverse_triples (Dataset attribute)
,
[1]
(EagerInductiveDataset attribute)
(InductiveDataset attribute)
create_random() (RankBasedMetricResults class method)
create_relation_to_entity_set_mapping() (in module pykeen.utils)
CrossE (class in pykeen.models)
CrossEInteraction (class in pykeen.nn.modules)
CrossEntropyLoss (class in pykeen.losses)
CSGraphAnchorSearcher (class in pykeen.nn.node_piece)
CSKG (class in pykeen.datasets)
CSVResultTracker (class in pykeen.trackers)
D
Dataset (class in pykeen.datasets)
(class in pykeen.datasets.base)
DATASET_NAME (CN3l attribute)
(WK3l120k attribute)
(WK3l15k attribute)
dataset_resolver (in module pykeen.datasets)
dataset_similarity() (in module pykeen.datasets.base)
DB100K (class in pykeen.datasets)
DBpedia50 (class in pykeen.datasets)
Decomposition (class in pykeen.nn.message_passing)
default_core_initializer() (TuckERInteraction static method)
default_core_initializer_kwargs (TuckERInteraction attribute)
default_entity_normalizer_kwargs (PairRE attribute)
DEFAULT_LOG_ADVERSARIAL_TEMPERATURE (InfoNCELoss attribute)
DegreeAnchorSelection (class in pykeen.nn.node_piece)
DeltaPointwiseLoss (class in pykeen.losses)
DerivedRankBasedMetric (class in pykeen.metrics.ranking)
deteriorate() (Dataset method)
,
[1]
(in module pykeen.triples.deteriorate)
DeterministicCleaner (class in pykeen.triples.splitting)
device (Model attribute)
(Representation attribute)
(TrainingLoop attribute)
DeviceHint (in module pykeen.typing)
diagonal_covariance (GaussianDistribution attribute)
dimensions (Interaction attribute)
DirectionAverageInteraction (class in pykeen.nn.modules)
DisjointGraphPairCombinator (class in pykeen.datasets.ea.combination)
DisjointInductivePathDataset (class in pykeen.datasets.inductive)
DistMA (class in pykeen.models)
DistMAInteraction (class in pykeen.nn.modules)
DistMult (class in pykeen.models)
DistMultInteraction (class in pykeen.nn.modules)
DistMultLiteral (class in pykeen.models)
DistMultLiteralGated (class in pykeen.models)
divisor (ModuloCheckpointKeeper attribute)
docdata() (Dataset class method)
,
[1]
DoubleMarginLoss (class in pykeen.losses)
DRKG (class in pykeen.datasets)
E
EagerDataset (class in pykeen.datasets.base)
EagerInductiveDataset (class in pykeen.datasets.inductive)
EarlyStopper (class in pykeen.stoppers)
EdgeWeighting (class in pykeen.nn.weighting)
einsum() (in module pykeen.utils)
Embedding (class in pykeen.nn.representation)
encode_all() (TextEncoder method)
end_run() (ConsoleResultTracker method)
(FileResultTracker method)
(MLFlowResultTracker method)
(MultiResultTracker method)
(ResultTracker method)
(TensorBoardResultTracker method)
(WANDBResultTracker method)
ensure_complex() (in module pykeen.utils)
ensure_ftp_directory() (in module pykeen.utils)
ensure_torch_random_state() (in module pykeen.utils)
ensure_tuple() (in module pykeen.utils)
entities_to_ids() (CoreTriplesFactory method)
(TriplesFactory method)
entity_constrainer_default_kwargs (HolE attribute)
entity_id_to_label (TriplesFactory attribute)
entity_relation_co_occurrence() (in module pykeen.triples.analysis)
entity_shape (BoxEInteraction attribute)
(ClampedInteraction attribute)
(ConvEInteraction attribute)
(CPInteraction attribute)
(Interaction attribute)
(KG2EInteraction attribute)
(MultiLinearTuckerInteraction attribute)
(MuREInteraction attribute)
(QuatEInteraction attribute)
(SimplEInteraction attribute)
(TransDInteraction attribute)
entity_to_id (Dataset attribute)
,
[1]
(TriplesFactory attribute)
entity_word_cloud() (TriplesFactory method)
env() (in module pykeen)
ERMLP (class in pykeen.models)
ERMLPE (class in pykeen.models)
ERMLPEInteraction (class in pykeen.nn.modules)
ERMLPInteraction (class in pykeen.nn.modules)
ERModel (class in pykeen.models)
estimate_cost_of_sequence() (in module pykeen.utils)
estimate_diversity() (NodePieceRepresentation method)
evaluate() (Evaluator method)
(OGBEvaluator method)
evaluation_batch_size (EarlyStopper attribute)
evaluation_loop (EvaluationLoopTrainingCallback attribute)
evaluation_slice_size (EarlyStopper attribute)
EvaluationLoopTrainingCallback (class in pykeen.training.callbacks)
EvaluationOnlyModel (class in pykeen.models)
EvaluationTrainingCallback (class in pykeen.training.callbacks)
Evaluator (class in pykeen.evaluation)
evaluator_resolver (in module pykeen.evaluation)
EveryCheckpointSchedule (class in pykeen.checkpoints)
exchange_df() (Predictions method)
ExpectationNormalizedMetric (class in pykeen.metrics.ranking)
expected_value() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
ExpectedLikelihood (class in pykeen.nn.sim)
ExplicitCheckpointKeeper (class in pykeen.checkpoints)
ExplicitCheckpointSchedule (class in pykeen.checkpoints)
extend() (AutoSFInteraction method)
extend_batch() (in module pykeen.utils)
extension (CSVResultTracker attribute)
(JSONResultTracker attribute)
extra_repr() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(ClassificationMetric method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(ExtraReprMixin method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(Metric method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
ExtraRelationGraphPairCombinator (class in pykeen.datasets.ea.combination)
ExtraReprMixin (class in pykeen.utils)
F
factor_sizes() (TensorTrainRepresentation class method)
factory_dict (Dataset attribute)
,
[1]
FB15k (class in pykeen.datasets)
FB15k237 (class in pykeen.datasets)
FeaturizedMessagePassingRepresentation (class in pykeen.nn.pyg)
FIGSHARE_LINK (OpenEA attribute)
file_name_entity_to_id (TriplesFactory attribute)
file_name_literal_to_id (TriplesNumericLiteralsFactory attribute)
file_name_numeric_literals (TriplesNumericLiteralsFactory attribute)
file_name_relation_to_id (TriplesFactory attribute)
FILE_NAMES (CN3l attribute)
(WK3l120k attribute)
(WK3l15k attribute)
FileResultTracker (class in pykeen.trackers)
filter_triples() (Predictions method)
filter_unique() (AnchorSelection method)
Filterer (class in pykeen.sampling.filtering)
filterer_resolver (in module pykeen.sampling.filtering)
final_checkpoints() (in module pykeen.checkpoints)
finalize() (ClassificationEvaluator method)
(Evaluator method)
(MacroRankBasedEvaluator method)
(RankBasedEvaluator method)
(ScoreConsumer method)
finalize_multi() (RankBasedEvaluator method)
finalize_with_confidence() (RankBasedEvaluator method)
fix_dataclass_init_docs() (in module pykeen.utils)
FixedModel (class in pykeen.models)
flatten (AllScoreConsumer attribute)
(TopKScoreConsumer attribute)
flatten_dictionary() (in module pykeen.utils)
FocalLoss (class in pykeen.losses)
format_relative_comparison() (in module pykeen.utils)
forward() (AdversarialLoss method)
(AttentionEdgeWeighting method)
(BCEAfterSigmoidLoss method)
(BCEWithLogitsLoss method)
(Bias method)
(BoxEInteraction method)
(ClampedInteraction method)
(ClassificationMetric method)
(Combination method)
(CombinedCompGCNRepresentations method)
(CombinedRegularizer method)
(CompGCNLayer method)
(ComplexSeparatedCombination method)
(CompositionModule method)
(ConcatAggregationCombination method)
(ConcatCombination method)
(ConcatMLP method)
(ConcatProjectionCombination method)
(ConvEInteraction method)
(ConvKBInteraction method)
(CPInteraction method)
(CrossEInteraction method)
(Decomposition method)
(DeltaPointwiseLoss method)
(DirectionAverageInteraction method)
(DoubleMarginLoss method)
(EdgeWeighting method)
(ERMLPEInteraction method)
(ERMLPInteraction method)
(ERModel method)
(ExpectedLikelihood method)
(Filterer method)
(FocalLoss method)
(FunctionalCompositionModule method)
(FunctionalInteraction method)
(GatedCombination method)
(Interaction method)
(InverseInDegreeEdgeWeighting method)
(InverseOutDegreeEdgeWeighting method)
(KG2EInteraction method)
(KG2ESimilarity method)
(LineaREInteraction method)
(LitModule method)
(LpRegularizer method)
(MarginPairwiseLoss method)
(MonotonicAffineTransformationInteraction method)
(MSELoss method)
(MultiLinearTuckerInteraction method)
(MuREInteraction method)
(NegativeKullbackLeiblerDivergence method)
(NoRegularizer method)
(NormLimitRegularizer method)
(NTNInteraction method)
(OrthogonalityRegularizer method)
(PairREInteraction method)
(PowerSumRegularizer method)
(ProjEInteraction method)
(QuatEInteraction method)
(Regularizer method)
(Representation method)
(RESCALInteraction method)
(RGCNLayer method)
(SEInteraction method)
(SymmetricEdgeWeighting method)
(TextEncoder method)
(TorusEInteraction method)
(TransDInteraction method)
(TransEInteraction method)
(TransFInteraction method)
(TransformerInteraction method)
(TransHInteraction method)
(TransRInteraction method)
(TripleREInteraction method)
(TuckERInteraction method)
(UMInteraction method)
forward_horizontally_stacked() (BasesDecomposition method)
(BlockDecomposition method)
(Decomposition method)
forward_normalized() (CharacterEmbeddingTextEncoder method)
(TextEncoder method)
(TransformerTextEncoder method)
forward_vertically_stacked() (BasesDecomposition method)
(BlockDecomposition method)
(Decomposition method)
frequency (EarlyStopper attribute)
(EveryCheckpointSchedule attribute)
from_dataset() (TextRepresentation class method)
(WikidataVisualRepresentation class method)
from_directory_binary() (Dataset class method)
,
[1]
from_labeled_triples() (TriplesFactory class method)
(TriplesNumericLiteralsFactory class method)
from_path() (Dataset class method)
,
[1]
(TriplesFactory class method)
(TriplesNumericLiteralsFactory class method)
from_path_binary() (CoreTriplesFactory class method)
from_ranks() (RankBasedMetricResults class method)
from_scores() (ClassificationMetricResults class method)
(UncertainPrediction class method)
from_searched_sf() (AutoSFInteraction class method)
from_tf() (Dataset static method)
,
[1]
from_tokenizer() (TokenizationRepresentation class method)
from_triples() (Instances class method)
(LCWAInstances class method)
(SLCWAInstances class method)
from_triples_factory() (CachedTextRepresentation class method)
(LabelBasedInitializer class method)
(TextRepresentation class method)
(WikidataVisualRepresentation class method)
func() (AutoSFInteraction static method)
(CircularCorrelationCompositionModule method)
(ComplExInteraction static method)
(DistMAInteraction static method)
(DistMultInteraction static method)
(HolEInteraction static method)
(MultiplicationCompositionModule method)
(RotatEInteraction static method)
(SubtractionCompositionModule method)
FunctionalCompositionModule (class in pykeen.nn.compositions)
FunctionalInteraction (class in pykeen.nn.modules)
G
GalkinPrecomputedTokenizerLoader (class in pykeen.nn.node_piece)
GatedCombination (class in pykeen.nn.combination)
GaussianDistribution (class in pykeen.typing)
generalized_harmonic_numbers() (in module pykeen.metrics.ranking)
generate_num_candidates_and_ranks() (in module pykeen.metrics.ranking)
generate_ranks() (in module pykeen.metrics.ranking)
generate_triples() (in module pykeen.triples.generation)
generate_triples_factory() (in module pykeen.triples.generation)
GeometricMeanRank (class in pykeen.metrics.ranking)
get_absolute_split_sizes() (in module pykeen.triples.splitting)
get_batchnorm_modules() (in module pykeen.utils)
get_benchmark() (in module pykeen.utils)
get_coefficients() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(ReindexedMetric method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
get_collator() (Instances method)
(LCWAEvaluationLoop method)
(SLCWAInstances method)
get_configuration() (MultiResultTracker method)
get_connected_components() (in module pykeen.utils)
get_dataset() (in module pykeen.datasets)
get_description() (AdjustedArithmeticMeanRank class method)
(AdjustedArithmeticMeanRankIndex class method)
(AdjustedGeometricMeanRankIndex class method)
(AdjustedHitsAtK class method)
(AdjustedInverseHarmonicMeanRank class method)
(ArithmeticMeanRank class method)
(ClassificationMetric class method)
(Count class method)
(DerivedRankBasedMetric class method)
(ExpectationNormalizedMetric class method)
(GeometricMeanRank class method)
(HarmonicMeanRank class method)
(HitsAtK class method)
(InverseArithmeticMeanRank class method)
(InverseGeometricMeanRank class method)
(InverseHarmonicMeanRank class method)
(InverseMedianRank class method)
(MedianRank class method)
(Metric class method)
(RankBasedMetric class method)
,
[1]
(ReindexedMetric class method)
(StandardDeviation class method)
(Variance class method)
(ZArithmeticMeanRank class method)
(ZGeometricMeanRank class method)
(ZHitsAtK class method)
(ZInverseHarmonicMeanRank class method)
(ZMetric class method)
get_descriptions() (WikidataTextCache method)
get_devices() (in module pykeen.utils)
get_df_io() (in module pykeen.utils)
get_dropout_modules() (in module pykeen.utils)
get_edge_index() (in module pykeen.utils)
get_entities() (in module pykeen.triples.utils)
get_entity_count_df() (in module pykeen.datasets.analysis)
get_entity_counts() (in module pykeen.triples.analysis)
get_entity_relation_co_occurrence_df() (in module pykeen.datasets.analysis)
get_expected_norm() (in module pykeen.utils)
get_git_branch() (in module pykeen.version)
get_git_hash() (in module pykeen.version)
get_grad_params() (Model method)
get_inverse_relation_id() (CoreTriplesFactory method)
(TriplesFactory method)
get_json_bytes_io() (in module pykeen.utils)
get_labels() (WikidataTextCache method)
get_last_lr() (LRScheduler method)
get_link() (AdjustedArithmeticMeanRank class method)
(AdjustedArithmeticMeanRankIndex class method)
(AdjustedGeometricMeanRankIndex class method)
(AdjustedHitsAtK class method)
(AdjustedInverseHarmonicMeanRank class method)
(ArithmeticMeanRank class method)
(ClassificationMetric class method)
(Count class method)
(DerivedRankBasedMetric class method)
(ExpectationNormalizedMetric class method)
(GeometricMeanRank class method)
(HarmonicMeanRank class method)
(HitsAtK class method)
(InverseArithmeticMeanRank class method)
(InverseGeometricMeanRank class method)
(InverseHarmonicMeanRank class method)
(InverseMedianRank class method)
(MedianRank class method)
(Metric class method)
(RankBasedMetric class method)
,
[1]
(ReindexedMetric class method)
(StandardDeviation class method)
(Variance class method)
(ZArithmeticMeanRank class method)
(ZGeometricMeanRank class method)
(ZHitsAtK class method)
(ZInverseHarmonicMeanRank class method)
(ZMetric class method)
get_lr() (LRScheduler method)
get_mapped_triples() (in module pykeen.triples)
get_mask_for_relations() (CoreTriplesFactory method)
(TriplesFactory method)
get_metric() (MetricResults method)
(PipelineResult method)
get_model_io() (in module pykeen.utils)
get_most_frequent_relations() (CoreTriplesFactory method)
get_normalized_name() (Dataset method)
,
[1]
(Evaluator class method)
(NegativeSampler class method)
(Regularizer class method)
(TrainingLoop class method)
get_numeric_literals_tensor() (TriplesNumericLiteralsFactory method)
get_optimal_sequence() (in module pykeen.utils)
get_preferred_device() (in module pykeen.utils)
get_range() (AdjustedArithmeticMeanRank class method)
(AdjustedArithmeticMeanRankIndex class method)
(AdjustedGeometricMeanRankIndex class method)
(AdjustedHitsAtK class method)
(AdjustedInverseHarmonicMeanRank class method)
(ArithmeticMeanRank class method)
(ClassificationMetric class method)
(Count class method)
(DerivedRankBasedMetric class method)
(ExpectationNormalizedMetric class method)
(GeometricMeanRank class method)
(HarmonicMeanRank class method)
(HitsAtK class method)
(InverseArithmeticMeanRank class method)
(InverseGeometricMeanRank class method)
(InverseHarmonicMeanRank class method)
(InverseMedianRank class method)
(MedianRank class method)
(Metric class method)
(RankBasedMetric class method)
,
[1]
(ReindexedMetric class method)
(StandardDeviation class method)
(Variance class method)
(ZArithmeticMeanRank class method)
(ZGeometricMeanRank class method)
(ZHitsAtK class method)
(ZInverseHarmonicMeanRank class method)
(ZMetric class method)
get_relation_cardinality_types_df() (in module pykeen.datasets.analysis)
get_relation_count_df() (in module pykeen.datasets.analysis)
get_relation_counts() (in module pykeen.triples.analysis)
get_relation_functionality() (in module pykeen.triples.analysis)
get_relation_functionality_df() (in module pykeen.datasets.analysis)
get_relation_pattern_types_df() (in module pykeen.datasets.analysis)
get_relations() (in module pykeen.triples.utils)
get_sampled_values() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
get_shapes_and_einsum_eq() (TensorTrainRepresentation static method)
get_summary_dict() (EarlyStopper method)
(NopStopper method)
(Stopper method)
get_texts() (IdentityCache method)
(PyOBOTextCache method)
(TextCache method)
(WikidataTextCache method)
get_until_first_blank() (in module pykeen.utils)
get_version() (in module pykeen.version)
Globi (class in pykeen.datasets)
GradientAbsClippingTrainingCallback (class in pykeen.training.callbacks)
GradientNormClippingTrainingCallback (class in pykeen.training.callbacks)
GraphPairCombinator (class in pykeen.datasets.ea.combination)
H
hamiltonian_product() (in module pykeen.nn.quaternion)
harmonic_variances() (in module pykeen.metrics.ranking)
HarmonicMeanRank (class in pykeen.metrics.ranking)
has_dataset() (in module pykeen.datasets)
HashDiversityInfo (class in pykeen.nn.node_piece)
head_column (PackedZipRemoteDataset attribute)
head_indices (Interaction attribute)
head_shape (Interaction attribute)
HEADER (CSVResultTracker attribute)
header (PackedZipRemoteDataset attribute)
HEADERS (WikidataTextCache attribute)
HeadRepresentation (class in pykeen.typing)
Hetionet (class in pykeen.datasets)
HitsAtK (class in pykeen.metrics.ranking)
HolE (class in pykeen.models)
HolEInteraction (class in pykeen.nn.modules)
hpo_default (AutoSF attribute)
(BoxE attribute)
(CombinedRegularizer attribute)
(CompGCN attribute)
(ComplEx attribute)
(ComplExLiteral attribute)
(ConvE attribute)
(ConvKB attribute)
(CooccurrenceFilteredModel attribute)
(CP attribute)
(CrossE attribute)
(DeltaPointwiseLoss attribute)
(DistMA attribute)
(DistMult attribute)
(DistMultLiteral attribute)
(DistMultLiteralGated attribute)
(DoubleMarginLoss attribute)
(ERMLP attribute)
(ERMLPE attribute)
(FixedModel attribute)
(HolE attribute)
(InductiveNodePiece attribute)
(InfoNCELoss attribute)
(KG2E attribute)
(Loss attribute)
(MarginPairwiseLoss attribute)
(MarginRankingLoss attribute)
(MuRE attribute)
(NegativeSampler attribute)
(NodePiece attribute)
(NoRegularizer attribute)
(NSSALoss attribute)
(NTN attribute)
(PairRE attribute)
(PairwiseLogisticLoss attribute)
(PointwiseHingeLoss attribute)
(ProjE attribute)
(QuatE attribute)
(Regularizer attribute)
(RESCAL attribute)
(RGCN attribute)
(RotatE attribute)
(SE attribute)
(SimplE attribute)
(SoftMarginRankingLoss attribute)
(SoftplusLoss attribute)
(SoftPointwiseHingeLoss attribute)
(TorusE attribute)
(TrainingLoop attribute)
(TransD attribute)
(TransE attribute)
(TransF attribute)
(TransH attribute)
(TransR attribute)
(TuckER attribute)
(UM attribute)
hpo_pipeline() (in module pykeen.hpo)
hpo_pipeline_from_config() (in module pykeen.hpo)
hpo_pipeline_from_path() (in module pykeen.hpo)
HpoPipelineResult (class in pykeen.hpo)
I
IdentityCache (class in pykeen.nn.text)
ILPC2022Large (class in pykeen.datasets.inductive)
ILPC2022Small (class in pykeen.datasets.inductive)
increasing (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(Count attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
index() (AffineTransformationParameters method)
inductive_inference (LazyInductiveDataset attribute)
inductive_testing (LazyInductiveDataset attribute)
inductive_validation (EagerInductiveDataset attribute)
(InductiveDataset attribute)
(LazyInductiveDataset attribute)
InductiveDataset (class in pykeen.datasets.inductive)
InductiveERModel (class in pykeen.models)
InductiveFB15k237 (class in pykeen.datasets.inductive)
InductiveMode (in module pykeen.typing)
InductiveNELL (class in pykeen.datasets.inductive)
InductiveNodePiece (class in pykeen.models)
InductiveNodePieceGNN (class in pykeen.models)
InductiveWN18RR (class in pykeen.datasets.inductive)
InfoNCELoss (class in pykeen.losses)
init_phases() (in module pykeen.nn.init)
Initializer (in module pykeen.typing)
Instances (class in pykeen.triples)
Interaction (class in pykeen.nn.modules)
interaction_resolver (in module pykeen.nn.modules)
InverseArithmeticMeanRank (class in pykeen.metrics.ranking)
InverseGeometricMeanRank (class in pykeen.metrics.ranking)
InverseHarmonicMeanRank (class in pykeen.metrics.ranking)
InverseInDegreeEdgeWeighting (class in pykeen.nn.weighting)
InverseMedianRank (class in pykeen.metrics.ranking)
InverseOutDegreeEdgeWeighting (class in pykeen.nn.weighting)
invert_mapping() (in module pykeen.utils)
is_complex (ComplExInteraction attribute)
(Interaction attribute)
(RotatEInteraction attribute)
is_triple_tensor_subset() (in module pykeen.utils)
isin_many_dim() (in module pykeen.utils)
iter_extra_repr() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(AnchorSelection method)
(ArithmeticMeanRank method)
(BasesDecomposition method)
(BlockDecomposition method)
(ClassificationMetric method)
(ConcatAggregationCombination method)
(ConcatCombination method)
(CoreTriplesFactory method)
(Count method)
(Dataset method)
,
[1]
(Decomposition method)
(DerivedRankBasedMetric method)
(EagerDataset method)
(ExpectationNormalizedMetric method)
(ExtraReprMixin method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(KGInfo method)
(MedianRank method)
(Metric method)
(MixtureAnchorSelection method)
(PageRankAnchorSelection method)
(PersonalizedPageRankAnchorSearcher method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(Representation method)
(ScipySparseAnchorSearcher method)
(SparseBFSSearcher method)
(StandardDeviation method)
(TensorTrainRepresentation method)
(TokenizationRepresentation method)
(TriplesNumericLiteralsFactory method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
J
JSONResultTracker (class in pykeen.trackers)
K
keep (LastCheckpointKeeper attribute)
key (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(ClassificationMetric attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(Metric attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
key_from_string() (ClassificationMetricResults class method)
(MetricResults class method)
(RankBasedMetricResults class method)
key_to_string() (MetricResults class method)
KG2E (class in pykeen.models)
kg2e_similarity_resolver (in module pykeen.nn.sim)
KG2EInteraction (class in pykeen.nn.modules)
KG2ESimilarity (class in pykeen.nn.sim)
KGInfo (class in pykeen.triples)
Kinships (class in pykeen.datasets)
L
label_triples() (TriplesFactory method)
LabelBasedInitializer (class in pykeen.nn.init)
LabeledTriples (in module pykeen.typing)
larger_is_better (EarlyStopper attribute)
LastCheckpointKeeper (class in pykeen.checkpoints)
latex_visualize() (AutoSFInteraction method)
LazyDataset (class in pykeen.datasets.base)
LazyInductiveDataset (class in pykeen.datasets.inductive)
LCWAEvaluationLoop (class in pykeen.evaluation)
LCWAInstances (class in pykeen.triples)
LCWALitModule (class in pykeen.contrib.lightning)
LCWATrainingLoop (class in pykeen.training)
LineaREInteraction (class in pykeen.nn.modules)
literal_shape (TriplesNumericLiteralsFactory attribute)
LiteralModel (class in pykeen.models)
LitModule (class in pykeen.contrib.lightning)
load_state() (Model method)
load_state_dict() (LRScheduler method)
load_summary_dict_from_training_loop_checkpoint() (Stopper static method)
load_triples() (in module pykeen.triples.utils)
log_metrics() (ConsoleResultTracker method)
(CSVResultTracker method)
(JSONResultTracker method)
(MLFlowResultTracker method)
(MultiResultTracker method)
(NeptuneResultTracker method)
(PythonResultTracker method)
(ResultTracker method)
(TensorBoardResultTracker method)
(WANDBResultTracker method)
log_params() (ConsoleResultTracker method)
(CSVResultTracker method)
(JSONResultTracker method)
(MLFlowResultTracker method)
(MultiResultTracker method)
(NeptuneResultTracker method)
(PythonResultTracker method)
(ResultTracker method)
(TensorBoardResultTracker method)
(WANDBResultTracker method)
logcumsumexp() (in module pykeen.utils)
Loss (class in pykeen.losses)
loss (TrainingCallback attribute)
(TrainingLoop attribute)
loss_default_kwargs (BoxE attribute)
(ComplEx attribute)
(ComplExLiteral attribute)
(ConvE attribute)
(DistMultLiteral attribute)
(DistMultLiteralGated attribute)
(ERMLPE attribute)
(Model attribute)
(PairRE attribute)
(ProjE attribute)
(QuatE attribute)
(SimplE attribute)
(TuckER attribute)
loss_resolver (in module pykeen.losses)
lower (ValueRange attribute)
lower_inclusive (ValueRange attribute)
LowRankRepresentation (class in pykeen.nn.representation)
lp_norm() (in module pykeen.utils)
LpRegularizer (class in pykeen.regularizers)
lr_scheduler_resolver (in module pykeen.lr_schedulers)
lr_schedulers_hpo_defaults (in module pykeen.lr_schedulers)
LRScheduler (class in pykeen.lr_schedulers)
M
MacroRankBasedEvaluator (class in pykeen.evaluation)
make_model() (in module pykeen.models)
make_model_cls() (in module pykeen.models)
map_triples() (TriplesFactory method)
mapped_triples (ProcessedTuple attribute)
MarginalDistributionBaseline (class in pykeen.models)
MarginPairwiseLoss (class in pykeen.losses)
MarginRankingLoss (class in pykeen.losses)
maximize (MetricSelection attribute)
mean (GaussianDistribution attribute)
MedianRank (class in pykeen.metrics.ranking)
message() (CompGCNLayer method)
MessagePassingRepresentation (class in pykeen.nn.pyg)
metadata (Dataset attribute)
,
[1]
metadata_file_name (Dataset attribute)
,
[1]
METADATA_FILE_NAME (PipelineResult attribute)
MetisAnchorTokenizer (class in pykeen.nn.node_piece)
Metric (class in pykeen.metrics)
metric (EarlyStopper attribute)
metric_resolver (in module pykeen.evaluation)
MetricResults (class in pykeen.evaluation)
metrics (ClassificationMetricResults attribute)
(RankBasedMetricResults attribute)
MetricSelection (class in pykeen.checkpoints)
MissingDropoutError
MixtureAnchorSelection (class in pykeen.nn.node_piece)
MLFlowResultTracker (class in pykeen.trackers)
Model (class in pykeen.models)
model (TrainingCallback attribute)
MODEL_FILE_NAME (PipelineResult attribute)
model_resolver (in module pykeen.models)
module
pykeen
pykeen.checkpoints
pykeen.constants
pykeen.contrib.lightning
pykeen.datasets
pykeen.datasets.analysis
pykeen.datasets.base
pykeen.datasets.ea.combination
pykeen.datasets.inductive
pykeen.evaluation
pykeen.losses
pykeen.lr_schedulers
pykeen.metrics
pykeen.metrics.ranking
pykeen.models
pykeen.models.multimodal
pykeen.models.resolve
pykeen.models.uncertainty
pykeen.models.unimodal
pykeen.nn
pykeen.nn.combination
pykeen.nn.compositions
pykeen.nn.init
pykeen.nn.message_passing
pykeen.nn.modules
pykeen.nn.node_piece
pykeen.nn.perceptron
pykeen.nn.pyg
pykeen.nn.quaternion
pykeen.nn.representation
pykeen.nn.sim
pykeen.nn.text
pykeen.nn.utils
pykeen.nn.vision
pykeen.nn.weighting
pykeen.optimizers
pykeen.pipeline
pykeen.pipeline.api
pykeen.predict
pykeen.regularizers
pykeen.sampling
pykeen.sampling.filtering
pykeen.stoppers
pykeen.trackers
pykeen.training
pykeen.training.callbacks
pykeen.triples
pykeen.triples.analysis
pykeen.triples.deteriorate
pykeen.triples.generation
pykeen.triples.leakage
pykeen.triples.remix
pykeen.triples.splitting
pykeen.triples.utils
pykeen.typing
pykeen.utils
pykeen.version
ModuloCheckpointKeeper (class in pykeen.checkpoints)
MonotonicAffineTransformationInteraction (class in pykeen.nn.modules)
MSELoss (class in pykeen.losses)
MultiLinearTuckerInteraction (class in pykeen.nn.modules)
multiplication_table() (in module pykeen.nn.quaternion)
MultiplicationCompositionModule (class in pykeen.nn.compositions)
MultiResultTracker (class in pykeen.trackers)
MultiTrainingCallback (class in pykeen.training.callbacks)
MuRE (class in pykeen.models)
MuREInteraction (class in pykeen.nn.modules)
Mutation (in module pykeen.typing)
N
name (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(Count attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(OGBBioKG attribute)
(OGBWikiKG2 attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
Nations (class in pykeen.datasets)
needs_candidates (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
needs_message (AttentionEdgeWeighting attribute)
(EdgeWeighting attribute)
negative_loss_term_unreduced() (AdversarialBCEWithLogitsLoss method)
(AdversarialLoss method)
(NSSALoss method)
negative_norm() (in module pykeen.utils)
negative_norm_of_sum() (in module pykeen.utils)
negative_sampler_resolver (in module pykeen.sampling)
NegativeKullbackLeiblerDivergence (class in pykeen.nn.sim)
NegativeSampler (class in pykeen.sampling)
NeptuneResultTracker (class in pykeen.trackers)
nested_get() (in module pykeen.utils)
new_with_restriction() (CoreTriplesFactory method)
(TriplesFactory method)
NoClosedFormError
NodePiece (class in pykeen.models)
NodePieceRepresentation (class in pykeen.nn.node_piece)
NonFiniteLossError
NopStopper (class in pykeen.stoppers)
NoRandomSeedNecessary (class in pykeen.utils)
NoRegularizer (class in pykeen.regularizers)
normalize (CombinedRegularizer attribute)
normalize() (in module pykeen.nn.quaternion)
normalize_path() (in module pykeen.utils)
normalize_rank_type() (in module pykeen.typing)
normalize_ratios() (in module pykeen.triples.splitting)
normalize_string() (in module pykeen.utils)
normalize_target() (in module pykeen.typing)
Normalizer (in module pykeen.typing)
NormBasedInteraction (class in pykeen.nn.modules)
NormLimitRegularizer (class in pykeen.regularizers)
notate() (ValueRange method)
NSSALoss (class in pykeen.losses)
NTN (class in pykeen.models)
NTNInteraction (class in pykeen.nn.modules)
num_bases (LowRankRepresentation attribute)
num_bits() (BloomFilterer static method)
num_entities (Dataset attribute)
,
[1]
num_parameter_bytes (Model attribute)
num_parameters (Model attribute)
num_probes() (BloomFilterer static method)
num_real_relations (Model attribute)
num_relations (Dataset attribute)
,
[1]
num_tokens (TokenizationRepresentation attribute)
num_triples (CoreTriplesFactory attribute)
number_results (EarlyStopper attribute)
numeric_expected_value() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
numeric_expected_value_with_ci() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
numeric_variance() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
numeric_variance_with_ci() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
O
objective (HpoPipelineResult attribute)
offset (AffineTransformationParameters attribute)
OGBBioKG (class in pykeen.datasets)
OGBEvaluator (class in pykeen.evaluation)
OGBWikiKG2 (class in pykeen.datasets)
on_batch() (MultiTrainingCallback method)
(TrainingCallback method)
on_before_zero_grad() (LitModule method)
OpenBioLink (class in pykeen.datasets)
OpenBioLinkLQ (class in pykeen.datasets)
OpenEA (class in pykeen.datasets)
optimizer (TrainingCallback attribute)
optimizer_resolver (in module pykeen.optimizers)
optimizers_hpo_defaults (in module pykeen.optimizers)
OrthogonalityRegularizer (class in pykeen.regularizers)
output_shape() (Combination method)
(ComplexSeparatedCombination method)
P
PackedZipRemoteDataset (class in pykeen.datasets.base)
PageRankAnchorSelection (class in pykeen.nn.node_piece)
PairRE (class in pykeen.models)
PairREInteraction (class in pykeen.nn.modules)
PairwiseLogisticLoss (class in pykeen.losses)
PairwiseLoss (class in pykeen.losses)
PartiallyRestrictedPredictionDataset (class in pykeen.predict)
PartitionRepresentation (class in pykeen.nn.representation)
pass_messages() (FeaturizedMessagePassingRepresentation method)
(MessagePassingRepresentation method)
(SimpleMessagePassingRepresentation method)
(TypedMessagePassingRepresentation method)
PathDataset (class in pykeen.datasets.base)
patience (EarlyStopper attribute)
PersonalizedPageRankAnchorSearcher (class in pykeen.nn.node_piece)
PharMeBINet (class in pykeen.datasets)
PharmKG (class in pykeen.datasets)
PharmKG8k (class in pykeen.datasets)
pipeline() (in module pykeen.pipeline)
pipeline_from_config() (in module pykeen.pipeline)
pipeline_from_path() (in module pykeen.pipeline)
PipelineResult (class in pykeen.pipeline)
plot() (in module pykeen.pipeline)
(PipelineResult method)
plot_early_stopping() (in module pykeen.pipeline)
(PipelineResult method)
plot_er() (in module pykeen.pipeline)
(PipelineResult method)
plot_losses() (in module pykeen.pipeline)
(PipelineResult method)
point_to_box_distance() (BoxEInteraction static method)
PointwiseHingeLoss (class in pykeen.losses)
PointwiseLoss (class in pykeen.losses)
pop_regularization_term() (Regularizer method)
positive_loss_term() (AdversarialBCEWithLogitsLoss method)
(AdversarialLoss method)
(NSSALoss method)
post_batch() (MultiTrainingCallback method)
(TrainingCallback method)
post_epoch() (CheckpointTrainingCallback method)
(EvaluationLoopTrainingCallback method)
(EvaluationTrainingCallback method)
(MultiTrainingCallback method)
(StopperTrainingCallback method)
(TrackerTrainingCallback method)
(TrainingCallback method)
post_forward_pass() (Model method)
post_parameter_update() (CombinedCompGCNRepresentations method)
(Embedding method)
(Model method)
(Regularizer method)
(Representation method)
(RGCNRepresentation method)
post_train() (MultiTrainingCallback method)
(TrainingCallback method)
powersum_norm() (in module pykeen.utils)
PowerSumRegularizer (class in pykeen.regularizers)
pre_batch() (MultiTrainingCallback method)
(TrainingCallback method)
pre_step() (GradientAbsClippingTrainingCallback method)
(GradientNormClippingTrainingCallback method)
(MultiTrainingCallback method)
(TrainingCallback method)
precalculate_anchor_ppr() (PersonalizedPageRankAnchorSearcher method)
PrecomputedPoolTokenizer (class in pykeen.nn.node_piece)
PrecomputedTokenizerLoader (class in pykeen.nn.node_piece)
predict() (Model method)
predict_all() (in module pykeen.predict)
predict_h() (CooccurrenceFilteredModel method)
(Model method)
predict_h_uncertain() (in module pykeen.models.uncertainty)
predict_hrt() (Model method)
predict_hrt_uncertain() (in module pykeen.models.uncertainty)
predict_r() (CooccurrenceFilteredModel method)
(Model method)
predict_r_uncertain() (in module pykeen.models.uncertainty)
predict_t() (CooccurrenceFilteredModel method)
(Model method)
predict_t_uncertain() (in module pykeen.models.uncertainty)
predict_target() (in module pykeen.predict)
predict_triples() (in module pykeen.predict)
predict_uncertain_helper() (in module pykeen.models.uncertainty)
PredictionDataset (class in pykeen.predict)
Predictions (class in pykeen.predict)
prefix (MetricSelection attribute)
prepare_ablation() (in module pykeen.ablation)
prepare_ablation_from_config() (in module pykeen.ablation)
prepare_filter_triples() (in module pykeen.utils)
PretrainedInitializer (class in pykeen.nn.init)
PrimeKG (class in pykeen.datasets)
print_lr() (LRScheduler method)
probe() (BloomFilterer method)
process() (CollapseGraphPairCombinator method)
(DisjointGraphPairCombinator method)
(ExtraRelationGraphPairCombinator method)
(GraphPairCombinator method)
(ScorePack method)
(SwapGraphPairCombinator method)
process_batch() (LCWAEvaluationLoop method)
process_lcwa_scores() (AdversarialLoss method)
(CrossEntropyLoss method)
(DoubleMarginLoss method)
(InfoNCELoss method)
(Loss method)
(MarginPairwiseLoss method)
process_scores_() (ClassificationEvaluator method)
(Evaluator method)
(MacroRankBasedEvaluator method)
(RankBasedEvaluator method)
(SampledRankBasedEvaluator method)
process_slcwa_scores() (AdversarialLoss method)
(CrossEntropyLoss method)
(DoubleMarginLoss method)
(InfoNCELoss method)
(Loss method)
(MarginPairwiseLoss method)
ProcessedTuple (class in pykeen.datasets.ea.combination)
product_normalize() (BoxEInteraction static method)
ProjE (class in pykeen.models)
project_entity() (in module pykeen.utils)
ProjEInteraction (class in pykeen.nn.modules)
PseudoTypedNegativeSampler (class in pykeen.sampling)
pykeen
module
pykeen.checkpoints
module
pykeen.constants
module
pykeen.contrib.lightning
module
pykeen.datasets
module
pykeen.datasets.analysis
module
pykeen.datasets.base
module
pykeen.datasets.ea.combination
module
pykeen.datasets.inductive
module
pykeen.evaluation
module
pykeen.losses
module
pykeen.lr_schedulers
module
pykeen.metrics
module
pykeen.metrics.ranking
module
pykeen.models
module
pykeen.models.multimodal
module
pykeen.models.resolve
module
pykeen.models.uncertainty
module
pykeen.models.unimodal
module
pykeen.nn
module
pykeen.nn.combination
module
pykeen.nn.compositions
module
pykeen.nn.init
module
pykeen.nn.message_passing
module
pykeen.nn.modules
module
pykeen.nn.node_piece
module
pykeen.nn.perceptron
module
pykeen.nn.pyg
module
pykeen.nn.quaternion
module
pykeen.nn.representation
module
pykeen.nn.sim
module
pykeen.nn.text
module
pykeen.nn.utils
module
pykeen.nn.vision
module
pykeen.nn.weighting
module
pykeen.optimizers
module
pykeen.pipeline
module
pykeen.pipeline.api
module
pykeen.predict
module
pykeen.regularizers
module
pykeen.sampling
module
pykeen.sampling.filtering
module
pykeen.stoppers
module
pykeen.trackers
module
pykeen.training
module
pykeen.training.callbacks
module
pykeen.triples
module
pykeen.triples.analysis
module
pykeen.triples.deteriorate
module
pykeen.triples.generation
module
pykeen.triples.leakage
module
pykeen.triples.remix
module
pykeen.triples.splitting
module
pykeen.triples.utils
module
pykeen.typing
module
pykeen.utils
module
pykeen.version
module
PYKEEN_BENCHMARKS (in module pykeen.constants)
PYKEEN_CHECKPOINTS (in module pykeen.constants)
PYKEEN_DATASETS (in module pykeen.constants)
PYKEEN_EXPERIMENTS (in module pykeen.constants)
PYKEEN_HOME (in module pykeen.constants)
PYKEEN_LOGS (in module pykeen.constants)
PyOBOTextCache (class in pykeen.nn.text)
PythonResultTracker (class in pykeen.trackers)
PythonSetFilterer (class in pykeen.sampling.filtering)
Q
QuatE (class in pykeen.models)
QuatEInteraction (class in pykeen.nn.modules)
query() (WikidataTextCache class method)
query_text() (WikidataTextCache class method)
R
random_non_negative_int() (in module pykeen.utils)
RandomAnchorSelection (class in pykeen.nn.node_piece)
RandomizedCleaner (class in pykeen.triples.splitting)
RandomWalkPositionalEncodingInitializer (class in pykeen.nn.init)
rank() (DegreeAnchorSelection method)
(PageRankAnchorSelection method)
(RandomAnchorSelection method)
(SingleSelection method)
RankBasedEvaluator (class in pykeen.evaluation)
RankBasedMetric (class in pykeen.metrics)
(class in pykeen.metrics.ranking)
RankBasedMetricResults (class in pykeen.evaluation)
rate_limited() (in module pykeen.utils)
ratios (CompressedSingleDataset attribute)
(SingleTabbedDataset attribute)
(TabbedDataset attribute)
register_callback() (MultiTrainingCallback method)
register_training_loop() (MultiTrainingCallback method)
(TrainingCallback method)
regularization_term (Regularizer attribute)
Regularizer (class in pykeen.regularizers)
regularizer_default_kwargs (ComplEx attribute)
(ConvKB attribute)
(DistMult attribute)
(QuatE attribute)
(RESCAL attribute)
(SimplE attribute)
(TransH attribute)
regularizer_resolver (in module pykeen.regularizers)
reindex() (in module pykeen.triples.leakage)
ReindexedMetric (class in pykeen.metrics.ranking)
relation_cardinality_types() (in module pykeen.triples.analysis)
relation_column (PackedZipRemoteDataset attribute)
relation_id_to_label (TriplesFactory attribute)
relation_injectivity() (in module pykeen.triples.analysis)
relation_pattern_types() (in module pykeen.triples.analysis)
relation_regularizer_default_kwargs (TransH attribute)
relation_shape (BoxEInteraction attribute)
(ClampedInteraction attribute)
(CPInteraction attribute)
(CrossEInteraction attribute)
(Interaction attribute)
(KG2EInteraction attribute)
(LineaREInteraction attribute)
(MultiLinearTuckerInteraction attribute)
(MuREInteraction attribute)
(NTNInteraction attribute)
(PairREInteraction attribute)
(QuatEInteraction attribute)
(RESCALInteraction attribute)
(SEInteraction attribute)
(SimplEInteraction attribute)
(TransDInteraction attribute)
(TransHInteraction attribute)
(TransRInteraction attribute)
(TripleREInteraction attribute)
(UMInteraction attribute)
relation_to_id (Dataset attribute)
,
[1]
(TriplesFactory attribute)
relation_word_cloud() (TriplesFactory method)
RelationRepresentation (class in pykeen.typing)
relations_to_ids() (CoreTriplesFactory method)
(TriplesFactory method)
RelationTokenizer (class in pykeen.nn.node_piece)
relative_delta (EarlyStopper attribute)
remaining_patience (EarlyStopper attribute)
remix() (Dataset method)
,
[1]
(in module pykeen.triples.remix)
RemoteDataset (class in pykeen.datasets.base)
replicate_best_pipeline() (HpoPipelineResult method)
replicate_pipeline_from_config() (in module pykeen.pipeline)
replicate_pipeline_from_path() (in module pykeen.pipeline)
Representation (class in pykeen.nn.representation)
RESCAL (class in pykeen.models)
RESCALInteraction (class in pykeen.nn.modules)
reset() (Regularizer method)
reset_parameters() (BasesDecomposition method)
(Bias method)
(BlockDecomposition method)
(CompGCNLayer method)
(ConvKBInteraction method)
(Decomposition method)
(Embedding method)
(ERMLPInteraction method)
(Interaction method)
(LowRankRepresentation method)
(MonotonicAffineTransformationInteraction method)
(MultiLinearTuckerInteraction method)
(ProjEInteraction method)
(Representation method)
(RGCNRepresentation method)
(TuckERInteraction method)
reset_parameters_() (InductiveNodePieceGNN method)
(Model method)
resolve_device() (in module pykeen.utils)
resolve_margin() (DoubleMarginLoss static method)
resolve_result_trackers() (in module pykeen.trackers)
restrict() (Dataset method)
,
[1]
Result (class in pykeen.utils)
RESULT_FILE_NAME (PipelineResult attribute)
result_tracker (EarlyStopper attribute)
(TrainingCallback attribute)
ResultTracker (class in pykeen.trackers)
RGCN (class in pykeen.models)
RGCNLayer (class in pykeen.nn.message_passing)
RGCNRepresentation (class in pykeen.nn.message_passing)
RotatE (class in pykeen.models)
RotatEInteraction (class in pykeen.nn.modules)
S
safe_diagonal() (in module pykeen.nn.utils)
sample() (NegativeSampler method)
SampledRankBasedEvaluator (class in pykeen.evaluation)
save() (TorchPrecomputedTokenizerLoader static method)
save_assignment() (TokenizationRepresentation method)
save_model() (in module pykeen.checkpoints)
(PipelineResult method)
save_state() (Model method)
save_to_directory() (HpoPipelineResult method)
(PipelineResult method)
(Result method)
save_to_ftp() (HpoPipelineResult method)
(PipelineResult method)
(Result method)
save_to_s3() (HpoPipelineResult method)
(PipelineResult method)
(Result method)
scale (AffineTransformationParameters attribute)
ScipySparseAnchorSearcher (class in pykeen.nn.node_piece)
score (UncertainPrediction attribute)
score() (Interaction method)
score_h() (CooccurrenceFilteredModel method)
(ERModel method)
(FixedModel method)
(Interaction method)
(MarginalDistributionBaseline method)
(Model method)
(SoftInverseTripleBaseline method)
score_h_inverse() (Model method)
score_hrt() (CooccurrenceFilteredModel method)
(ERModel method)
(EvaluationOnlyModel method)
(FixedModel method)
(Interaction method)
(Model method)
score_hrt_inverse() (Model method)
score_r() (CooccurrenceFilteredModel method)
(ERModel method)
(EvaluationOnlyModel method)
(FixedModel method)
(Interaction method)
(Model method)
score_t() (CooccurrenceFilteredModel method)
(ERModel method)
(FixedModel method)
(Interaction method)
(MarginalDistributionBaseline method)
(Model method)
(SoftInverseTripleBaseline method)
score_t_inverse() (Model method)
ScoreConsumer (class in pykeen.predict)
ScorePack (class in pykeen.predict)
SE (class in pykeen.models)
Sealant (class in pykeen.triples.leakage)
SEInteraction (class in pykeen.nn.modules)
select() (ScipySparseAnchorSearcher static method)
(SparseBFSSearcher static method)
sep (PackedZipRemoteDataset attribute)
set_random_seed() (in module pykeen.utils)
SetwiseLoss (class in pykeen.losses)
SHA512 (OpenEA attribute)
ShapeError
should_evaluate() (EarlyStopper method)
(NopStopper method)
(Stopper method)
should_stop() (EarlyStopper method)
(NopStopper method)
(Stopper method)
similarity() (Dataset method)
,
[1]
SimplE (class in pykeen.models)
SimplEInteraction (class in pykeen.nn.modules)
SimpleMessagePassingRepresentation (class in pykeen.nn.pyg)
simulate_checkpoints() (in module pykeen.checkpoints)
SingleCompGCNRepresentation (class in pykeen.nn.representation)
SingleSelection (class in pykeen.nn.node_piece)
SingleTabbedDataset (class in pykeen.datasets.base)
SLCWAInstances (class in pykeen.triples)
SLCWALitModule (class in pykeen.contrib.lightning)
SLCWATrainingLoop (class in pykeen.training)
SoftInverseTripleBaseline (class in pykeen.models)
SoftMarginRankingLoss (class in pykeen.losses)
SoftplusLoss (class in pykeen.losses)
SoftPointwiseHingeLoss (class in pykeen.losses)
SparseBFSSearcher (class in pykeen.nn.node_piece)
split() (CombinedCompGCNRepresentations method)
(CoreTriplesFactory method)
(in module pykeen.triples.splitting)
(Splitter method)
split_absolute_size() (CleanupSplitter method)
(CoverageSplitter method)
(Splitter method)
split_complex() (in module pykeen.utils)
split_workload() (in module pykeen.utils)
Splitter (class in pykeen.triples.splitting)
StandardDeviation (class in pykeen.metrics.ranking)
start_run() (ConsoleResultTracker method)
(CSVResultTracker method)
(MLFlowResultTracker method)
(MultiResultTracker method)
(PythonResultTracker method)
(ResultTracker method)
(WANDBResultTracker method)
state_dict() (LRScheduler method)
std() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
step() (LRScheduler method)
stopped (EarlyStopper attribute)
Stopper (class in pykeen.stoppers)
stopper (PipelineResult attribute)
stopper_resolver (in module pykeen.stoppers)
StopperTrainingCallback (class in pykeen.training.callbacks)
string_or_key_to_key() (MetricResults class method)
study (HpoPipelineResult attribute)
sub_batch_and_slice() (TrainingLoop method)
SubsetRepresentation (class in pykeen.nn.representation)
SubtractionCompositionModule (class in pykeen.nn.compositions)
summarize() (Dataset method)
,
[1]
(InductiveDataset method)
summary_str() (Dataset method)
,
[1]
(InductiveDataset method)
supported_rank_types (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
supports_slicing (LCWATrainingLoop attribute)
(TrainingLoop attribute)
supports_weights (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(ClassificationMetric attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(Metric attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
SwapGraphPairCombinator (class in pykeen.datasets.ea.combination)
SymmetricEdgeWeighting (class in pykeen.nn.weighting)
SymmetricLCWATrainingLoop (class in pykeen.training)
synonyms (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(BCEWithLogitsLoss attribute)
(ClassificationMetric attribute)
(Count attribute)
(DerivedRankBasedMetric attribute)
(ExpectationNormalizedMetric attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(Loss attribute)
(MarginRankingLoss attribute)
(MedianRank attribute)
(Metric attribute)
(MSELoss attribute)
(NSSALoss attribute)
(RankBasedMetric attribute)
,
[1]
(ReindexedMetric attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
T
TabbedDataset (class in pykeen.datasets.base)
tail_column (PackedZipRemoteDataset attribute)
tail_indices (Interaction attribute)
tail_shape (Interaction attribute)
TailRepresentation (class in pykeen.typing)
TarFileRemoteDataset (class in pykeen.datasets.base)
TarFileSingleDataset (class in pykeen.datasets.base)
Target (in module pykeen.typing)
TargetColumn (in module pykeen.typing)
TargetPredictions (class in pykeen.predict)
tensor_product() (in module pykeen.utils)
tensor_sum() (in module pykeen.utils)
tensor_to_df() (CoreTriplesFactory method)
(in module pykeen.triples.utils)
(TriplesFactory method)
TensorBoardResultTracker (class in pykeen.trackers)
TensorTrainRepresentation (class in pykeen.nn.representation)
term (Regularizer property)
testing (LazyDataset attribute)
TextCache (class in pykeen.nn.text)
TextEncoder (class in pykeen.nn.text)
TextRepresentation (class in pykeen.nn.representation)
title (PipelineResult attribute)
to_core_triples_factory() (TriplesFactory method)
to_df() (MetricResults method)
to_dict() (MetricResults method)
to_directory_binary() (Dataset method)
,
[1]
to_flat_dict() (MetricResults method)
to_path_binary() (CoreTriplesFactory method)
(TriplesFactory method)
(TriplesNumericLiteralsFactory method)
TokenizationRepresentation (class in pykeen.nn.node_piece)
Tokenizer (class in pykeen.nn.node_piece)
topk_argpartition() (CSGraphAnchorSearcher static method)
topk_argsort() (CSGraphAnchorSearcher static method)
TopKScoreConsumer (class in pykeen.predict)
TorchPrecomputedTokenizerLoader (class in pykeen.nn.node_piece)
TorchRandomHint (in module pykeen.typing)
TorusE (class in pykeen.models)
TorusEInteraction (class in pykeen.nn.modules)
TrackerTrainingCallback (class in pykeen.training.callbacks)
train() (CombinedCompGCNRepresentations method)
(TrainingLoop method)
train_dataloader() (LitModule method)
training (LazyDataset attribute)
training_loop (TrainingCallback attribute)
training_loop_resolver (in module pykeen.training)
training_step() (LitModule method)
TRAINING_TRIPLES_FILE_NAME (PipelineResult attribute)
TrainingCallback (class in pykeen.training.callbacks)
TrainingCallbackHint (in module pykeen.training.callbacks)
TrainingLoop (class in pykeen.training)
TransD (class in pykeen.models)
TransDInteraction (class in pykeen.nn.modules)
transductive_training (LazyInductiveDataset attribute)
TransE (class in pykeen.models)
TransEInteraction (class in pykeen.nn.modules)
TransF (class in pykeen.models)
TransFInteraction (class in pykeen.nn.modules)
TransformedRepresentation (class in pykeen.nn.representation)
TransformerInteraction (class in pykeen.nn.modules)
TransformerTextEncoder (class in pykeen.nn.text)
TransH (class in pykeen.models)
TransHInteraction (class in pykeen.nn.modules)
translation_kwargs (ProcessedTuple attribute)
TransR (class in pykeen.models)
TransRInteraction (class in pykeen.nn.modules)
triple_tensor_to_set() (in module pykeen.utils)
TripleCoverageError
TriplePredictions (class in pykeen.predict)
TripleREInteraction (class in pykeen.nn.modules)
triples (TriplesFactory attribute)
triples_file_name (CoreTriplesFactory attribute)
triples_pair_sort_key() (Dataset class method)
,
[1]
triples_sort_key() (Dataset static method)
,
[1]
TriplesFactory (class in pykeen.triples)
TriplesNumericLiteralsFactory (class in pykeen.triples)
TuckER (class in pykeen.models)
TuckERInteraction (class in pykeen.nn.modules)
TypedMessagePassingRepresentation (class in pykeen.nn.pyg)
U
UM (class in pykeen.models)
UMInteraction (class in pykeen.nn.modules)
UMLS (class in pykeen.datasets)
UncertainPrediction (class in pykeen.models.uncertainty)
uncertainty (UncertainPrediction attribute)
UnionCheckpointKeeper (class in pykeen.checkpoints)
UnionCheckpointSchedule (class in pykeen.checkpoints)
uniques_per_representation (HashDiversityInfo attribute)
uniques_total (HashDiversityInfo attribute)
unleak() (in module pykeen.triples.leakage)
unpack_singletons() (in module pykeen.utils)
UnpackedRemoteDataset (class in pykeen.datasets.base)
UnpackedRemoteDisjointInductiveDataset (class in pykeen.datasets.inductive)
update() (NoRegularizer method)
(OrthogonalityRegularizer method)
(Regularizer method)
updated (Regularizer attribute)
upgrade_to_sequence() (in module pykeen.utils)
upper (ValueRange attribute)
upper_inclusive (ValueRange attribute)
use_horizontal_stacking() (in module pykeen.nn.utils)
use_tqdm (EarlyStopper attribute)
V
val_dataloader() (LitModule method)
validate_labels() (PointwiseLoss static method)
validation (LazyDataset attribute)
validation_step() (LitModule method)
value_range (AdjustedArithmeticMeanRank attribute)
(AdjustedArithmeticMeanRankIndex attribute)
(AdjustedGeometricMeanRankIndex attribute)
(AdjustedHitsAtK attribute)
(AdjustedInverseHarmonicMeanRank attribute)
(ArithmeticMeanRank attribute)
(Count attribute)
(GeometricMeanRank attribute)
(HarmonicMeanRank attribute)
(HitsAtK attribute)
(Interaction attribute)
(InverseArithmeticMeanRank attribute)
(InverseGeometricMeanRank attribute)
(InverseHarmonicMeanRank attribute)
(InverseMedianRank attribute)
(MedianRank attribute)
(StandardDeviation attribute)
(Variance attribute)
(ZArithmeticMeanRank attribute)
(ZGeometricMeanRank attribute)
(ZHitsAtK attribute)
(ZInverseHarmonicMeanRank attribute)
(ZMetric attribute)
ValueRange (class in pykeen.metrics)
Variance (class in pykeen.metrics.ranking)
variance() (AdjustedArithmeticMeanRank method)
(AdjustedArithmeticMeanRankIndex method)
(AdjustedGeometricMeanRankIndex method)
(AdjustedHitsAtK method)
(AdjustedInverseHarmonicMeanRank method)
(ArithmeticMeanRank method)
(Count method)
(DerivedRankBasedMetric method)
(ExpectationNormalizedMetric method)
(GeometricMeanRank method)
(HarmonicMeanRank method)
(HitsAtK method)
(InverseArithmeticMeanRank method)
(InverseGeometricMeanRank method)
(InverseHarmonicMeanRank method)
(InverseMedianRank method)
(MedianRank method)
(RankBasedMetric method)
,
[1]
(ReindexedMetric method)
(StandardDeviation method)
(Variance method)
(ZArithmeticMeanRank method)
(ZGeometricMeanRank method)
(ZHitsAtK method)
(ZInverseHarmonicMeanRank method)
(ZMetric method)
verify() (ShapeError class method)
verify_ids() (WikidataTextCache static method)
view_complex() (in module pykeen.utils)
VisionDataset (class in pykeen.nn.vision)
VisualRepresentation (class in pykeen.nn.vision)
W
WANDBResultTracker (class in pykeen.trackers)
WD50KT (class in pykeen.datasets)
weight (Regularizer attribute)
WeisfeilerLehmanInitializer (class in pykeen.nn.init)
Wikidata5M (class in pykeen.datasets)
WIKIDATA_ENDPOINT (WikidataTextCache attribute)
WikidataTextCache (class in pykeen.nn.text)
WikidataTextRepresentation (class in pykeen.nn.representation)
WikidataVisualRepresentation (class in pykeen.nn.vision)
with_labels() (CoreTriplesFactory method)
WK3l120k (class in pykeen.datasets)
WK3l15k (class in pykeen.datasets)
WN18 (class in pykeen.datasets)
WN18RR (class in pykeen.datasets)
X
xavier_normal_() (in module pykeen.nn.init)
xavier_uniform_() (in module pykeen.nn.init)
Y
YAGO310 (class in pykeen.datasets)
Z
ZArithmeticMeanRank (class in pykeen.metrics.ranking)
ZGeometricMeanRank (class in pykeen.metrics.ranking)
ZHitsAtK (class in pykeen.metrics.ranking)
ZInverseHarmonicMeanRank (class in pykeen.metrics.ranking)
ZipSingleDataset (class in pykeen.datasets.base)
ZMetric (class in pykeen.metrics.ranking)