pykeen
v1.5.0
Getting Started
Installation
First Steps
Tracking Results during Training
Saving Checkpoints during Training
A Toy Example with Translational Distance Models
Understanding the Evaluation
Optimizing a Model’s Hyper-parameters
Novel Link Prediction
Running an Ablation Study
Performance Tricks
Bring Your Own
Bring Your Own Data
Bring Your Own Interaction
Extending PyKEEN
Extending the Datasets
Extending the Models
Extending the Interaction Models (Old-Style)
Reference
Pipeline
Models
Datasets
Triples
Training
Stoppers
Loss Functions
Regularizers
Result Trackers
Negative Sampling
Filtering
Evaluation
Hyper-parameter Optimization
Ablation
Lookup
Prediction
Sealant
Constants
pykeen.nn
Utilities
Appendix
References
pykeen
»
Index
Edit on GitHub
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
_
_NewAbstractModel (class in pykeen.models)
_OldAbstractModel (class in pykeen.models)
A
ablation_pipeline() (in module pykeen.ablation)
add() (BloomFilterer method)
add_parameter() (Regularizer method)
all_in_bounds() (in module pykeen.utils)
append_weight_regularizer() (ERModel method)
apply() (Sealant method)
apply_only_once (Regularizer attribute)
B
BasesDecomposition (class in pykeen.nn.message_passing)
BasicNegativeSampler (class in pykeen.sampling)
batch_and_slice() (Evaluator method)
batch_size_search() (LCWATrainingLoop method)
(SLCWATrainingLoop method)
(TrainingLoop method)
BCEAfterSigmoidLoss (class in pykeen.losses)
BCEWithLogitsLoss (class in pykeen.losses)
BernoulliNegativeSampler (class in pykeen.sampling)
best_epoch (EarlyStopper attribute)
best_metric (EarlyStopper attribute)
Bias (class in pykeen.utils)
BlockDecomposition (class in pykeen.nn.message_passing)
BloomFilterer (class in pykeen.sampling.filtering)
broadcast_cat() (in module pykeen.utils)
C
calculate_broadcasted_elementwise_result_shape() (in module pykeen.utils)
can_slice_h (Model attribute)
can_slice_r (Model attribute)
can_slice_t (Model attribute)
cast_constrainer() (in module pykeen.typing)
check_shapes() (in module pykeen.utils)
checksum (LCWATrainingLoop property)
(SLCWATrainingLoop property)
(TrainingLoop property)
CKG (class in pykeen.datasets)
clamp_norm() (in module pykeen.utils)
cli() (Dataset class method)
clone_and_exchange_triples() (CoreTriplesFactory method)
(TriplesFactory method)
CoDExLarge (class in pykeen.datasets)
CoDExMedium (class in pykeen.datasets)
CoDExSmall (class in pykeen.datasets)
collect_regularization_term() (_NewAbstractModel method)
(_OldAbstractModel method)
(Model method)
Combination (class in pykeen.nn.combinations)
combine_complex() (in module pykeen.utils)
CombinedRegularizer (class in pykeen.regularizers)
compact_mapping() (in module pykeen.utils)
CompGCN (class in pykeen.models)
ComplEx (class in pykeen.models)
complex_interaction() (in module pykeen.nn.functional)
complex_normalize() (in module pykeen.utils)
ComplexCombination (class in pykeen.nn.combinations)
ComplExInteraction (class in pykeen.nn.modules)
ComplExLiteral (class in pykeen.models)
ComplExLiteralCombination (class in pykeen.nn.combinations)
compose (class in pykeen.utils)
compressed (LCWAInstances attribute)
ConceptNet (class in pykeen.datasets)
ConsoleResultTracker (class in pykeen.trackers)
Constrainer (in module pykeen.typing)
constrainer_default_kwargs (KG2E attribute)
contains() (BloomFilterer method)
(Filterer method)
(PythonSetFilterer method)
continue_callbacks (EarlyStopper attribute)
ConvE (class in pykeen.models)
conve_interaction() (in module pykeen.nn.functional)
ConvEInteraction (class in pykeen.nn.modules)
convert_to_canonical_shape() (in module pykeen.utils)
ConvKB (class in pykeen.models)
convkb_interaction() (in module pykeen.nn.functional)
ConvKBInteraction (class in pykeen.nn.modules)
CoreTriplesFactory (class in pykeen.triples)
corrupt_batch() (BasicNegativeSampler method)
(BernoulliNegativeSampler method)
(NegativeSampler method)
(PseudoTypedNegativeSampler method)
Countries (class in pykeen.datasets)
create() (CoreTriplesFactory class method)
create_lcwa_instances() (CoreTriplesFactory method)
(TriplesNumericLiteralsFactory method)
create_slcwa_instances() (CoreTriplesFactory method)
(TriplesNumericLiteralsFactory method)
cross_e_interaction() (in module pykeen.nn.functional)
CrossE (class in pykeen.models)
CrossEInteraction (class in pykeen.nn.modules)
CrossEntropyLoss (class in pykeen.losses)
CSKG (class in pykeen.datasets)
CSVResultTracker (class in pykeen.trackers)
D
Dataset (class in pykeen.datasets.base)
DATASET_NAME (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_entity_normalizer_kwargs (PairRE attribute)
deteriorate() (Dataset method)
(in module pykeen.triples.deteriorate)
device (LCWATrainingLoop property)
(SLCWATrainingLoop property)
(TrainingLoop property)
DeviceHint (in module pykeen.typing)
diagonal_covariance (GaussianDistribution property)
DistMult (class in pykeen.models)
distmult_interaction() (in module pykeen.nn.functional)
DistMultCombination (class in pykeen.nn.combinations)
DistMultInteraction (class in pykeen.nn.modules)
DistMultLiteral (class in pykeen.models)
DRKG (class in pykeen.datasets)
E
EagerDataset (class in pykeen.datasets.base)
EarlyStopper (class in pykeen.stoppers)
EdgeWeighting (class in pykeen.nn.weighting)
Embedding (class in pykeen.nn.emb)
embedding_dim (Embedding property)
(EntityEmbeddingModel attribute)
(EntityRelationEmbeddingModel attribute)
(RepresentationModule property)
EmbeddingSpecification (class in pykeen.nn.emb)
end_run() (ConsoleResultTracker method)
(FileResultTracker method)
(MLFlowResultTracker method)
(ResultTracker method)
(TensorBoardResultTracker method)
ensure_ftp_directory() (in module pykeen.utils)
ensure_torch_random_state() (in module pykeen.utils)
ensure_tuple() (in module pykeen.utils)
entities_to_ids() (TriplesFactory method)
entity_constrainer_default_kwargs (HolE attribute)
entity_id_to_label (TriplesFactory property)
entity_representations (EntityEmbeddingModel attribute)
(EntityRelationEmbeddingModel attribute)
entity_shape (Interaction attribute)
(KG2EInteraction attribute)
(MuREInteraction attribute)
(SimplEInteraction attribute)
(TransDInteraction attribute)
entity_to_id (Dataset attribute)
(TriplesFactory property)
entity_word_cloud() (TriplesFactory method)
EntityEmbeddingModel (class in pykeen.models)
EntityRelationEmbeddingModel (class in pykeen.models)
env() (in module pykeen)
ERMLP (class in pykeen.models)
ermlp_interaction() (in module pykeen.nn.functional)
ERMLPE (class in pykeen.models)
ermlpe_interaction() (in module pykeen.nn.functional)
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)
evaluate() (Evaluator method)
(in module pykeen.evaluation)
evaluation_batch_size (EarlyStopper attribute)
evaluation_slice_size (EarlyStopper attribute)
evaluation_triples_factory (EarlyStopper attribute)
Evaluator (class in pykeen.evaluation)
evaluator (EarlyStopper attribute)
evaluator_resolver (in module pykeen.evaluation)
expected_likelihood() (in module pykeen.nn.sim)
(KG2E static method)
extend_batch() (in module pykeen.utils)
extended_einsum() (in module pykeen.utils)
extension (CSVResultTracker attribute)
(JSONResultTracker attribute)
extra_repr() (CoreTriplesFactory method)
(TriplesNumericLiteralsFactory method)
F
factory_dict (Dataset attribute)
FB15k (class in pykeen.datasets)
FB15k237 (class in pykeen.datasets)
FILE_NAMES (WK3l15k attribute)
FileResultTracker (class in pykeen.trackers)
Filterer (class in pykeen.sampling.filtering)
finalize() (Evaluator method)
(RankBasedEvaluator method)
(SklearnEvaluator method)
fix_dataclass_init_docs() (in module pykeen.utils)
flatten_dictionary() (in module pykeen.utils)
format_relative_comparison() (in module pykeen.utils)
forward() (_NewAbstractModel method)
(BasesDecomposition method)
(BCEAfterSigmoidLoss method)
(BCEWithLogitsLoss method)
(Bias method)
(BlockDecomposition method)
(Combination method)
(CombinedRegularizer method)
(ComplEx method)
(ComplexCombination method)
(CrossEntropyLoss method)
(Decomposition method)
(EdgeWeighting method)
(Embedding method)
(ERModel method)
(Filterer method)
(FunctionalInteraction method)
(Interaction method)
(InverseInDegreeEdgeWeighting method)
(InverseOutDegreeEdgeWeighting method)
(LiteralInteraction method)
(LpRegularizer method)
(MarginRankingLoss method)
(MonotonicAffineTransformationInteraction method)
(MSELoss method)
(NoRegularizer method)
(NSSALoss method)
(PowerSumRegularizer method)
(RealCombination method)
(Regularizer method)
(RepresentationModule method)
(SoftplusLoss method)
(SymmetricEdgeWeighting method)
(TransHRegularizer method)
frequency (EarlyStopper attribute)
from_dict() (RankBasedMetricResults class method)
(SklearnMetricResults class method)
from_json() (RankBasedMetricResults class method)
(SklearnMetricResults class method)
from_labeled_triples() (TriplesFactory class method)
from_path() (Dataset class method)
(TriplesFactory class method)
from_scores() (SklearnMetricResults class method)
from_tf() (Dataset static method)
from_triples() (Instances class method)
(LCWAInstances class method)
(SLCWAInstances class method)
func() (ComplExInteraction method)
(ConvEInteraction method)
(ConvKBInteraction method)
(CrossEInteraction method)
(DistMultInteraction method)
(ERMLPEInteraction method)
(ERMLPInteraction method)
(HolEInteraction method)
(KG2EInteraction method)
(MuREInteraction method)
(NTNInteraction method)
(PairREInteraction method)
(ProjEInteraction method)
(RESCALInteraction method)
(RotatEInteraction method)
(SimplEInteraction method)
(StructuredEmbeddingInteraction method)
(TransDInteraction method)
(TransEInteraction method)
(TransHInteraction method)
(TransRInteraction method)
(TuckerInteraction method)
(UnstructuredModelInteraction method)
FunctionalInteraction (class in pykeen.nn.modules)
G
GaussianDistribution (class in pykeen.typing)
get_all_prediction_df() (in module pykeen.models.predict)
(Model method)
get_batchnorm_modules() (in module pykeen.utils)
get_benchmark() (in module pykeen.utils)
get_dataset() (in module pykeen.datasets)
get_df_io() (in module pykeen.utils)
get_dimensions() (Interaction class method)
get_entities() (in module pykeen.triples.utils)
get_entity_count_df() (in module pykeen.datasets.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_head_prediction_df() (in module pykeen.models.predict)
(Model method)
get_in_canonical_shape() (RepresentationModule method)
get_in_more_canonical_shape() (RepresentationModule method)
get_inverse_relation_id() (CoreTriplesFactory method)
(TriplesFactory method)
get_json_bytes_io() (in module pykeen.utils)
get_mask_for_entities() (CoreTriplesFactory method)
(TriplesFactory method)
get_mask_for_relations() (CoreTriplesFactory method)
(TriplesFactory method)
get_metric() (MetricResults method)
(PipelineResult method)
(RankBasedMetricResults method)
(SklearnMetricResults method)
get_metric_list() (in module pykeen.evaluation)
get_model_io() (in module pykeen.utils)
get_most_frequent_relations() (CoreTriplesFactory method)
get_normalized_name() (Dataset class method)
(Evaluator class method)
(LCWATrainingLoop class method)
(NegativeSampler class method)
(Regularizer class method)
(SLCWATrainingLoop class method)
(TrainingLoop class method)
get_numeric_literals_tensor() (TriplesNumericLiteralsFactory method)
get_optimal_sequence() (in module pykeen.utils)
get_relation_cardinality_types_df() (in module pykeen.datasets.analysis)
get_relation_count_df() (in module pykeen.datasets.analysis)
get_relation_functionality_df() (in module pykeen.datasets.analysis)
get_relation_pattern_types_df() (in module pykeen.datasets.analysis)
get_relation_prediction_df() (in module pykeen.models.predict)
(Model method)
get_relations() (in module pykeen.triples.utils)
get_summary_dict() (EarlyStopper method)
(NopStopper method)
(Stopper method)
get_tail_prediction_df() (in module pykeen.models.predict)
(Model method)
get_until_first_blank() (in module pykeen.utils)
get_version() (in module pykeen.version)
H
has_dataset() (in module pykeen.datasets)
head_column (PackedZipRemoteDataset attribute)
HEADER (CSVResultTracker attribute)
header (PackedZipRemoteDataset attribute)
HeadRepresentation (in module pykeen.typing)
Hetionet (class in pykeen.datasets)
HolE (class in pykeen.models)
hole_interaction() (in module pykeen.nn.functional)
HolEInteraction (class in pykeen.nn.modules)
hpo_default (CompGCN attribute)
(ComplEx attribute)
(ComplExLiteral attribute)
(ConvE attribute)
(ConvKB attribute)
(CrossE attribute)
(DistMult attribute)
(DistMultLiteral attribute)
(ERMLP attribute)
(ERMLPE attribute)
(HolE attribute)
(KG2E attribute)
(LpRegularizer attribute)
(MarginRankingLoss attribute)
(MuRE attribute)
(NegativeSampler attribute)
(NoRegularizer attribute)
(NSSALoss attribute)
(NTN attribute)
(PairRE attribute)
(PowerSumRegularizer attribute)
(ProjE attribute)
(QuatE attribute)
(Regularizer attribute)
(RESCAL attribute)
(RGCN attribute)
(RotatE attribute)
(SimplE attribute)
(StructuredEmbedding attribute)
(TransD attribute)
(TransE attribute)
(TransH attribute)
(TransHRegularizer attribute)
(TransR attribute)
(TuckER attribute)
(UnstructuredModel 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
init_phases() (in module pykeen.nn.init)
init_with_device() (Embedding class method)
Initializer (in module pykeen.typing)
Instances (class in pykeen.triples)
Interaction (class in pykeen.nn.modules)
interaction_function() (ComplEx static method)
(DistMult static method)
(HolE static method)
(RotatE static method)
(TransD static method)
(TransR static method)
InverseInDegreeEdgeWeighting (class in pykeen.nn.weighting)
InverseOutDegreeEdgeWeighting (class in pykeen.nn.weighting)
invert_mapping() (in module pykeen.utils)
is_cuda_oom_error() (in module pykeen.utils)
is_cudnn_error() (in module pykeen.utils)
J
JSONResultTracker (class in pykeen.trackers)
K
KG2E (class in pykeen.models)
kg2e_interaction() (in module pykeen.nn.functional)
KG2EInteraction (class in pykeen.nn.modules)
Kinships (class in pykeen.datasets)
kullback_leibler_similarity() (in module pykeen.nn.sim)
(KG2E static method)
L
label_triples() (TriplesFactory method)
LabeledTriples (in module pykeen.typing)
larger_is_better (EarlyStopper attribute)
LazyDataset (class in pykeen.datasets.base)
LCWAInstances (class in pykeen.triples)
LCWATrainingLoop (class in pykeen.training)
LinearDropout (class in pykeen.nn.combinations)
LiteralInteraction (class in pykeen.nn.modules)
LiteralModel (class in pykeen.models)
LiteralRepresentation (class in pykeen.nn.emb)
load_state() (Model 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)
(NeptuneResultTracker method)
(ResultTracker method)
(TensorBoardResultTracker method)
(WANDBResultTracker method)
log_params() (ConsoleResultTracker method)
(CSVResultTracker method)
(JSONResultTracker method)
(MLFlowResultTracker method)
(NeptuneResultTracker method)
(ResultTracker method)
(TensorBoardResultTracker method)
(WANDBResultTracker method)
loss (LCWATrainingLoop property)
(SLCWATrainingLoop property)
(TrainingCallback attribute)
(TrainingLoop property)
loss_default_kwargs (ComplEx attribute)
(ComplExLiteral attribute)
(ConvE attribute)
(DistMultLiteral attribute)
(ERMLPE attribute)
(Model attribute)
(ProjE attribute)
(QuatE attribute)
(SimplE attribute)
(TuckER attribute)
loss_resolver (in module pykeen.losses)
lp_norm() (in module pykeen.utils)
LpRegularizer (class in pykeen.regularizers)
M
make() (EmbeddingSpecification method)
make_model() (in module pykeen.models)
make_model_cls() (in module pykeen.models)
mapped_triples (SLCWAInstances attribute)
MarginRankingLoss (class in pykeen.losses)
max_id (RepresentationModule attribute)
mean (GaussianDistribution property)
metric (EarlyStopper attribute)
metric_resolver (in module pykeen.evaluation)
MetricResults (class in pykeen.evaluation)
MLFlowResultTracker (class in pykeen.trackers)
Model (class in pykeen.models)
model (EarlyStopper attribute)
(TrainingCallback attribute)
model_resolver (in module pykeen.models)
module
pykeen
pykeen.constants
pykeen.datasets
pykeen.datasets.analysis
pykeen.datasets.base
pykeen.evaluation
pykeen.losses
pykeen.models
pykeen.models.multimodal
pykeen.models.predict
pykeen.models.resolve
pykeen.models.unimodal
pykeen.nn
pykeen.nn.combinations
pykeen.nn.emb
pykeen.nn.functional
pykeen.nn.init
pykeen.nn.message_passing
pykeen.nn.modules
pykeen.nn.sim
pykeen.nn.weighting
pykeen.pipeline
pykeen.pipeline.api
pykeen.regularizers
pykeen.sampling
pykeen.sampling.filtering
pykeen.stoppers
pykeen.trackers
pykeen.training.callbacks
pykeen.triples
pykeen.triples.deteriorate
pykeen.triples.leakage
pykeen.triples.remix
pykeen.triples.utils
pykeen.typing
pykeen.utils
pykeen.version
MonotonicAffineTransformationInteraction (class in pykeen.nn.modules)
MSELoss (class in pykeen.losses)
MultimodalInstances (class in pykeen.triples)
MultimodalLCWAInstances (class in pykeen.triples)
MultimodalSLCWAInstances (class in pykeen.triples)
MultiTrainingCallback (class in pykeen.training.callbacks)
MuRE (class in pykeen.models)
mure_interaction() (in module pykeen.nn.functional)
MuREInteraction (class in pykeen.nn.modules)
Mutation (in module pykeen.typing)
N
name (OGBBioKG attribute)
(OGBWikiKG attribute)
Nations (class in pykeen.datasets)
negative_norm() (in module pykeen.utils)
negative_norm_of_sum() (in module pykeen.utils)
negative_sampler_resolver (in module pykeen.sampling)
NegativeSampler (class in pykeen.sampling)
NeptuneResultTracker (class in pykeen.trackers)
new_with_restriction() (CoreTriplesFactory method)
(TriplesFactory method)
NopStopper (class in pykeen.stoppers)
NoRandomSeedNecessary (class in pykeen.utils)
NoRegularizer (class in pykeen.regularizers)
normalize (CombinedRegularizer attribute)
normalize_string() (in module pykeen.utils)
Normalizer (in module pykeen.typing)
NSSALoss (class in pykeen.losses)
NTN (class in pykeen.models)
ntn_interaction() (in module pykeen.nn.functional)
NTNInteraction (class in pykeen.nn.modules)
num_bits() (BloomFilterer static method)
num_embeddings (Embedding property)
num_entities (CoreTriplesFactory property)
(Dataset attribute)
num_parameter_bytes (Model attribute)
num_probes() (BloomFilterer static method)
num_relations (CoreTriplesFactory property)
(Dataset attribute)
num_triples (CoreTriplesFactory property)
number_results (EarlyStopper property)
numeric_literals (MultimodalInstances attribute)
O
objective (HpoPipelineResult attribute)
OGBBioKG (class in pykeen.datasets)
OGBWikiKG (class in pykeen.datasets)
on_batch() (MultiTrainingCallback method)
(TrainingCallback method)
OpenBioLink (class in pykeen.datasets)
OpenBioLinkLQ (class in pykeen.datasets)
optimizer (TrainingCallback attribute)
optimizer_resolver (in module pykeen.optimizers)
P
PackedZipRemoteDataset (class in pykeen.datasets.base)
pair_re_interaction() (in module pykeen.nn.functional)
PairRE (class in pykeen.models)
PairREInteraction (class in pykeen.nn.modules)
pairs (LCWAInstances attribute)
PairwiseLoss (class in pykeen.losses)
ParameterizedComplexCombination (class in pykeen.nn.combinations)
ParameterizedRealCombination (class in pykeen.nn.combinations)
PathDataset (class in pykeen.datasets.base)
patience (EarlyStopper attribute)
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)
PointwiseLoss (class in pykeen.losses)
pop_regularization_term() (Regularizer method)
post_batch() (MultiTrainingCallback method)
(TrainingCallback method)
post_epoch() (MultiTrainingCallback method)
(TrackerCallback method)
(TrainingCallback method)
post_forward_pass() (_OldAbstractModel method)
(Model method)
post_parameter_update() (_NewAbstractModel method)
(_OldAbstractModel method)
(Embedding method)
(EntityEmbeddingModel method)
(EntityRelationEmbeddingModel method)
(KG2E method)
(Model method)
(RepresentationModule method)
(TransH method)
post_train() (MultiTrainingCallback method)
(TrainingCallback method)
powersum_norm() (in module pykeen.utils)
PowerSumRegularizer (class in pykeen.regularizers)
predict() (in module pykeen.models.predict)
predict_h() (Model method)
predict_hrt() (Model method)
predict_r() (Model method)
predict_t() (Model method)
prepare_ablation() (in module pykeen.ablation)
prepare_ablation_from_config() (in module pykeen.ablation)
probe() (BloomFilterer method)
process_head_scores_() (Evaluator method)
(RankBasedEvaluator method)
(SklearnEvaluator method)
process_lcwa_scores() (MarginRankingLoss method)
(NSSALoss method)
process_slcwa_scores() (MarginRankingLoss method)
(NSSALoss method)
process_tail_scores_() (Evaluator method)
(RankBasedEvaluator method)
(SklearnEvaluator method)
ProjE (class in pykeen.models)
proje_interaction() (in module pykeen.nn.functional)
project_entity() (in module pykeen.utils)
ProjEInteraction (class in pykeen.nn.modules)
pruner_resolver (in module pykeen.hpo.pruners)
PseudoTypedNegativeSampler (class in pykeen.sampling)
pykeen
module
pykeen.constants
module
pykeen.datasets
module
pykeen.datasets.analysis
module
pykeen.datasets.base
module
pykeen.evaluation
module
pykeen.losses
module
pykeen.models
module
pykeen.models.multimodal
module
pykeen.models.predict
module
pykeen.models.resolve
module
pykeen.models.unimodal
module
pykeen.nn
module
pykeen.nn.combinations
module
pykeen.nn.emb
module
pykeen.nn.functional
module
pykeen.nn.init
module
pykeen.nn.message_passing
module
pykeen.nn.modules
module
pykeen.nn.sim
module
pykeen.nn.weighting
module
pykeen.pipeline
module
pykeen.pipeline.api
module
pykeen.regularizers
module
pykeen.sampling
module
pykeen.sampling.filtering
module
pykeen.stoppers
module
pykeen.trackers
module
pykeen.training.callbacks
module
pykeen.triples
module
pykeen.triples.deteriorate
module
pykeen.triples.leakage
module
pykeen.triples.remix
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)
PythonSetFilterer (class in pykeen.sampling.filtering)
Q
QuatE (class in pykeen.models)
R
random_non_negative_int() (in module pykeen.utils)
RankBasedEvaluator (class in pykeen.evaluation)
RankBasedMetricResults (class in pykeen.evaluation)
ratios (SingleTabbedDataset attribute)
(TabbedDataset attribute)
(TarFileSingleDataset attribute)
real_num_relations (CoreTriplesFactory property)
RealCombination (class in pykeen.nn.combinations)
register_callback() (MultiTrainingCallback method)
register_training_loop() (MultiTrainingCallback method)
(TrainingCallback method)
regularization_term (Regularizer attribute)
regularize_if_necessary() (_OldAbstractModel method)
(TransH method)
Regularizer (class in pykeen.regularizers)
regularizer_default (_NewAbstractModel attribute)
(_OldAbstractModel attribute)
regularizer_default_kwargs (_NewAbstractModel attribute)
(_OldAbstractModel attribute)
(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)
relation_column (PackedZipRemoteDataset attribute)
relation_dim (EntityRelationEmbeddingModel attribute)
relation_id_to_label (TriplesFactory property)
relation_representations (EntityRelationEmbeddingModel attribute)
relation_shape (CrossEInteraction attribute)
(Interaction attribute)
(KG2EInteraction attribute)
(MuREInteraction attribute)
(NTNInteraction attribute)
(PairREInteraction attribute)
(RESCALInteraction attribute)
(SimplEInteraction attribute)
(StructuredEmbeddingInteraction attribute)
(TransDInteraction attribute)
(TransHInteraction attribute)
(TransRInteraction attribute)
(UnstructuredModelInteraction attribute)
relation_to_id (Dataset attribute)
(TriplesFactory property)
relation_word_cloud() (TriplesFactory method)
RelationRepresentation (in module pykeen.typing)
relations_to_ids() (TriplesFactory method)
relative_delta (EarlyStopper attribute)
remaining_patience (EarlyStopper attribute)
remix() (Dataset method)
(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)
RepresentationModule (class in pykeen.nn.emb)
RESCAL (class in pykeen.models)
rescal_interaction() (in module pykeen.nn.functional)
RESCALInteraction (class in pykeen.nn.modules)
reset() (Regularizer method)
reset_parameters() (BasesDecomposition method)
(Bias method)
(BlockDecomposition method)
(ConvKBInteraction method)
(Decomposition method)
(Embedding method)
(ERMLPInteraction method)
(Interaction method)
(MonotonicAffineTransformationInteraction method)
(ProjEInteraction method)
(RepresentationModule method)
(TuckerInteraction method)
reset_parameters_() (Model method)
resolve_device() (in module pykeen.utils)
Result (class in pykeen.utils)
result (ScorePack property)
result_callbacks (EarlyStopper attribute)
result_tracker (EarlyStopper attribute)
results (EarlyStopper attribute)
ResultTracker (class in pykeen.trackers)
RGCN (class in pykeen.models)
RotatE (class in pykeen.models)
rotate_interaction() (in module pykeen.nn.functional)
RotatEInteraction (class in pykeen.nn.modules)
S
sample() (NegativeSampler method)
sampler_resolver (in module pykeen.hpo.samplers)
save_model() (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)
schema() (RankBasedMetricResults class method)
(SklearnMetricResults class method)
score() (Interaction method)
(ParameterizedRealCombination method)
(RealCombination method)
score_h() (_NewAbstractModel method)
(_OldAbstractModel method)
(ComplEx method)
(ConvE method)
(DistMult method)
(ERMLP method)
(ERMLPE method)
(HolE method)
(Interaction method)
(KG2E method)
(Model method)
(NTN method)
(ProjE method)
(RESCAL method)
(RotatE method)
(SimplE method)
(StructuredEmbedding method)
(TransD method)
(TransE method)
(TransH method)
(TransR method)
(TuckER method)
score_h_inverse() (Model method)
score_hrt() (_NewAbstractModel method)
(ComplEx method)
(ConvE method)
(ConvKB method)
(DistMult method)
(ERMLP method)
(ERMLPE method)
(HolE method)
(Interaction method)
(KG2E method)
(Model method)
(NTN method)
(ProjE method)
(RESCAL method)
(RotatE method)
(SimplE method)
(StructuredEmbedding method)
(TransD method)
(TransE method)
(TransH method)
(TransR method)
(TuckER method)
score_hrt_inverse() (Model method)
score_imag() (ComplexCombination method)
(ParameterizedComplexCombination method)
score_r() (_NewAbstractModel method)
(_OldAbstractModel method)
(ComplEx method)
(Interaction method)
(Model method)
score_real() (ComplexCombination method)
(ParameterizedComplexCombination method)
score_t() (_NewAbstractModel method)
(_OldAbstractModel method)
(ComplEx method)
(ConvE method)
(DistMult method)
(ERMLP method)
(ERMLPE method)
(HolE method)
(Interaction method)
(KG2E method)
(Model method)
(NTN method)
(ProjE method)
(RESCAL method)
(RotatE method)
(SimplE method)
(StructuredEmbedding method)
(TransD method)
(TransE method)
(TransH method)
(TransR method)
(TuckER method)
score_t_inverse() (Model method)
ScorePack (class in pykeen.typing)
scores (ScorePack property)
Sealant (class in pykeen.triples.leakage)
sep (PackedZipRemoteDataset attribute)
set_random_seed() (in module pykeen.utils)
SetwiseLoss (class in pykeen.losses)
shape (RepresentationModule attribute)
should_evaluate() (EarlyStopper method)
(NopStopper method)
(Stopper method)
should_stop() (EarlyStopper method)
(NopStopper method)
(Stopper method)
similarity() (Dataset method)
SimplE (class in pykeen.models)
simple_interaction() (in module pykeen.nn.functional)
SimplEInteraction (class in pykeen.nn.modules)
SingleTabbedDataset (class in pykeen.datasets.base)
SklearnEvaluator (class in pykeen.evaluation)
SklearnMetricResults (class in pykeen.evaluation)
SLCWAInstances (class in pykeen.triples)
SLCWATrainingLoop (class in pykeen.training)
SoftplusLoss (class in pykeen.losses)
split() (CoreTriplesFactory method)
split_complex() (in module pykeen.utils)
split_list_in_batches_iter() (in module pykeen.utils)
start_run() (ConsoleResultTracker method)
(CSVResultTracker method)
(MLFlowResultTracker method)
(ResultTracker method)
stopped (EarlyStopper attribute)
stopped_callbacks (EarlyStopper attribute)
Stopper (class in pykeen.stoppers)
stopper (PipelineResult attribute)
stopper_resolver (in module pykeen.stoppers)
strip_dim() (in module pykeen.utils)
structured_embedding_interaction() (in module pykeen.nn.functional)
StructuredEmbedding (class in pykeen.models)
StructuredEmbeddingInteraction (class in pykeen.nn.modules)
study (HpoPipelineResult attribute)
sub_batch_and_slice() (LCWATrainingLoop method)
(SLCWATrainingLoop method)
(TrainingLoop method)
summarize() (Dataset method)
summary_str() (Dataset method)
SymmetricEdgeWeighting (class in pykeen.nn.weighting)
synonyms (BCEWithLogitsLoss attribute)
(MarginRankingLoss attribute)
(MSELoss attribute)
(NSSALoss attribute)
T
TabbedDataset (class in pykeen.datasets.base)
tail_column (PackedZipRemoteDataset attribute)
tail_entity_shape (ConvEInteraction attribute)
(Interaction attribute)
TailRepresentation (in module pykeen.typing)
TarFileRemoteDataset (class in pykeen.datasets.base)
TarFileSingleDataset (class in pykeen.datasets.base)
tensor_product() (in module pykeen.utils)
tensor_sum() (in module pykeen.utils)
tensor_to_df() (CoreTriplesFactory method)
(TriplesFactory method)
TensorBoardResultTracker (class in pykeen.trackers)
term (Regularizer property)
testing (LazyDataset attribute)
title (PipelineResult attribute)
to_device_() (Model method)
to_df() (RankBasedMetricResults method)
to_dict() (RankBasedMetricResults method)
(SklearnMetricResults method)
to_flat_dict() (MetricResults method)
(RankBasedMetricResults method)
to_json() (RankBasedMetricResults method)
(SklearnMetricResults method)
torch_is_in_1d() (in module pykeen.utils)
TorchRandomHint (in module pykeen.typing)
TrackerCallback (class in pykeen.training.callbacks)
train() (LCWATrainingLoop method)
(SLCWATrainingLoop method)
(TrainingLoop method)
training (LazyDataset attribute)
training_loop (TrainingCallback attribute)
training_loop_resolver (in module pykeen.training)
training_triples_factory (EarlyStopper attribute)
TrainingCallback (class in pykeen.training.callbacks)
TrainingLoop (class in pykeen.training)
TransD (class in pykeen.models)
transd_interaction() (in module pykeen.nn.functional)
TransDInteraction (class in pykeen.nn.modules)
TransE (class in pykeen.models)
transe_interaction() (in module pykeen.nn.functional)
TransEInteraction (class in pykeen.nn.modules)
TransH (class in pykeen.models)
transh_interaction() (in module pykeen.nn.functional)
TransHInteraction (class in pykeen.nn.modules)
TransHRegularizer (class in pykeen.regularizers)
TranslationalInteraction (class in pykeen.nn.modules)
TransR (class in pykeen.models)
transr_interaction() (in module pykeen.nn.functional)
TransRInteraction (class in pykeen.nn.modules)
triples (TriplesFactory property)
TriplesFactory (class in pykeen.triples)
TriplesNumericLiteralsFactory (class in pykeen.triples)
TuckER (class in pykeen.models)
tucker_interaction() (in module pykeen.nn.functional)
TuckerInteraction (class in pykeen.nn.modules)
U
UMLS (class in pykeen.datasets)
unleak() (in module pykeen.triples.leakage)
unpack_singletons() (in module pykeen.utils)
UnpackedRemoteDataset (class in pykeen.datasets.base)
unstructured_model_interaction() (in module pykeen.nn.functional)
UnstructuredModel (class in pykeen.models)
UnstructuredModelInteraction (class in pykeen.nn.modules)
update() (NoRegularizer method)
(Regularizer method)
(TransHRegularizer method)
updated (Regularizer attribute)
upgrade_to_sequence() (in module pykeen.utils)
V
validate_labels() (PointwiseLoss static method)
validation (LazyDataset attribute)
view_complex() (in module pykeen.utils)
W
WANDBResultTracker (class in pykeen.trackers)
weight (Regularizer attribute)
with_labels() (CoreTriplesFactory method)
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)
Read the Docs
v: v1.5.0
Versions
latest
stable
v1.5.0
v1.4.0
v1.3.0
v1.1.0
v1.0.0
Downloads
On Read the Docs
Project Home
Builds