pykeen.nn

PyKEEN internal “nn” module.

nn

• Functional
  • Functions
    • boxe_interaction
      • boxe_interaction()
    • conve_interaction
      • conve_interaction()
    • convkb_interaction
      • convkb_interaction()
    • cp_interaction
      • cp_interaction()
    • cross_e_interaction
      • cross_e_interaction()
    • dist_ma_interaction
      • dist_ma_interaction()
    • distmult_interaction
      • distmult_interaction()
    • ermlp_interaction
      • ermlp_interaction()
    • ermlpe_interaction
      • ermlpe_interaction()
    • hole_interaction
      • hole_interaction()
    • kg2e_interaction
      • kg2e_interaction()
    • multilinear_tucker_interaction
      • multilinear_tucker_interaction()
    • mure_interaction
      • mure_interaction()
    • ntn_interaction
      • ntn_interaction()
    • pair_re_interaction
      • pair_re_interaction()
    • proje_interaction
      • proje_interaction()
    • rescal_interaction
      • rescal_interaction()
    • rotate_interaction
      • rotate_interaction()
    • simple_interaction
      • simple_interaction()
    • se_interaction
      • se_interaction()
    • transd_interaction
      • transd_interaction()
    • transe_interaction
      • transe_interaction()
    • transf_interaction
      • transf_interaction()
    • transh_interaction
      • transh_interaction()
    • transr_interaction
      • transr_interaction()
    • transformer_interaction
      • transformer_interaction()
    • triple_re_interaction
      • triple_re_interaction()
    • tucker_interaction
      • tucker_interaction()
    • um_interaction
      • um_interaction()
    • linea_re_interaction
      • linea_re_interaction()
• Stateful Interaction Modules
  • Classes
    • Interaction
      • Interaction
        • Interaction.entity_shape
        • Interaction.relation_shape
        • Interaction.tail_entity_shape
        • Interaction.value_range
        • Interaction.forward()
        • Interaction.full_entity_shapes()
        • Interaction.get_dimensions()
        • Interaction.head_indices()
        • Interaction.reset_parameters()
        • Interaction.score()
        • Interaction.score_h()
        • Interaction.score_hrt()
        • Interaction.score_r()
        • Interaction.score_t()
        • Interaction.tail_indices()
    • FunctionalInteraction
      • FunctionalInteraction
        • FunctionalInteraction.forward()
    • NormBasedInteraction
      • NormBasedInteraction
    • MonotonicAffineTransformationInteraction
      • MonotonicAffineTransformationInteraction
        • MonotonicAffineTransformationInteraction.forward()
        • MonotonicAffineTransformationInteraction.reset_parameters()
    • AutoSFInteraction
      • AutoSFInteraction
        • AutoSFInteraction.extend()
        • AutoSFInteraction.from_searched_sf()
        • AutoSFInteraction.func()
        • AutoSFInteraction.latex_visualize()
    • BoxEInteraction
      • BoxEInteraction
        • BoxEInteraction.entity_shape
        • BoxEInteraction.relation_shape
        • BoxEInteraction.func()
    • ComplExInteraction
      • ComplExInteraction
        • ComplExInteraction.func()
    • ConvEInteraction
      • ConvEInteraction
        • ConvEInteraction.tail_entity_shape
        • ConvEInteraction.func()
    • ConvKBInteraction
      • ConvKBInteraction
        • ConvKBInteraction.func()
        • ConvKBInteraction.reset_parameters()
    • CPInteraction
      • CPInteraction
        • CPInteraction.entity_shape
        • CPInteraction.relation_shape
        • CPInteraction.func()
    • CrossEInteraction
      • CrossEInteraction
        • CrossEInteraction.relation_shape
        • CrossEInteraction.func()
    • DistMAInteraction
      • DistMAInteraction
        • DistMAInteraction.func()
    • DistMultInteraction
      • DistMultInteraction
        • DistMultInteraction.func()
    • ERMLPEInteraction
      • ERMLPEInteraction
        • ERMLPEInteraction.func()
    • ERMLPInteraction
      • ERMLPInteraction
        • ERMLPInteraction.func()
        • ERMLPInteraction.reset_parameters()
    • HolEInteraction
      • HolEInteraction
        • HolEInteraction.func()
    • KG2EInteraction
      • KG2EInteraction
        • KG2EInteraction.entity_shape
        • KG2EInteraction.relation_shape
        • KG2EInteraction.func()
    • LineaREInteraction
      • LineaREInteraction
        • LineaREInteraction.relation_shape
        • LineaREInteraction.func()
    • MultiLinearTuckerInteraction
      • MultiLinearTuckerInteraction
        • MultiLinearTuckerInteraction.entity_shape
        • MultiLinearTuckerInteraction.relation_shape
        • MultiLinearTuckerInteraction.func()
        • MultiLinearTuckerInteraction.reset_parameters()
    • MuREInteraction
      • MuREInteraction
        • MuREInteraction.entity_shape
        • MuREInteraction.relation_shape
        • MuREInteraction.func()
    • NTNInteraction
      • NTNInteraction
        • NTNInteraction.relation_shape
        • NTNInteraction.func()
    • PairREInteraction
      • PairREInteraction
        • PairREInteraction.relation_shape
        • PairREInteraction.func()
    • ProjEInteraction
      • ProjEInteraction
        • ProjEInteraction.func()
        • ProjEInteraction.reset_parameters()
    • QuatEInteraction
      • QuatEInteraction
        • QuatEInteraction.entity_shape
        • QuatEInteraction.relation_shape
        • QuatEInteraction.func()
    • RESCALInteraction
      • RESCALInteraction
        • RESCALInteraction.relation_shape
        • RESCALInteraction.func()
    • RotatEInteraction
      • RotatEInteraction
        • RotatEInteraction.func()
    • SEInteraction
      • SEInteraction
        • SEInteraction.relation_shape
        • SEInteraction.func()
    • SimplEInteraction
      • SimplEInteraction
        • SimplEInteraction.entity_shape
        • SimplEInteraction.relation_shape
        • SimplEInteraction.func()
    • TorusEInteraction
      • TorusEInteraction
        • TorusEInteraction.func()
    • TransDInteraction
      • TransDInteraction
        • TransDInteraction.entity_shape
        • TransDInteraction.relation_shape
        • TransDInteraction.func()
    • TransEInteraction
      • TransEInteraction
        • TransEInteraction.func()
    • TransFInteraction
      • TransFInteraction
        • TransFInteraction.func()
    • TransformerInteraction
      • TransformerInteraction
        • TransformerInteraction.func()
    • TransHInteraction
      • TransHInteraction
        • TransHInteraction.relation_shape
        • TransHInteraction.func()
    • TransRInteraction
      • TransRInteraction
        • TransRInteraction.relation_shape
        • TransRInteraction.func()
    • TripleREInteraction
      • TripleREInteraction
        • TripleREInteraction.relation_shape
        • TripleREInteraction.func()
    • TuckerInteraction
      • TuckerInteraction
        • TuckerInteraction.default_core_initializer_kwargs
        • TuckerInteraction.default_core_initializer()
        • TuckerInteraction.func()
        • TuckerInteraction.reset_parameters()
    • UMInteraction
      • UMInteraction
        • UMInteraction.relation_shape
        • UMInteraction.func()
  • Class Inheritance Diagram
• Similarity
  • pykeen.nn.sim Module
    • Functions
      • expected_likelihood
        • expected_likelihood()
      • kullback_leibler_similarity
        • kullback_leibler_similarity()
• Representation
  • Classes
    • Representation
      • Representation
        • Representation.device
        • Representation.forward()
        • Representation.iter_extra_repr()
        • Representation.post_parameter_update()
        • Representation.reset_parameters()
    • Embedding
      • Embedding
        • Embedding.post_parameter_update()
        • Embedding.reset_parameters()
    • LowRankRepresentation
      • LowRankRepresentation
        • LowRankRepresentation.num_bases
        • LowRankRepresentation.approximate()
        • LowRankRepresentation.reset_parameters()
    • CompGCNLayer
      • CompGCNLayer
        • CompGCNLayer.forward()
        • CompGCNLayer.message()
        • CompGCNLayer.reset_parameters()
    • CombinedCompGCNRepresentations
      • CombinedCompGCNRepresentations
        • CombinedCompGCNRepresentations.forward()
        • CombinedCompGCNRepresentations.post_parameter_update()
        • CombinedCompGCNRepresentations.split()
        • CombinedCompGCNRepresentations.train()
    • PartitionRepresentation
      • PartitionRepresentation
    • BackfillRepresentation
      • BackfillRepresentation
    • SingleCompGCNRepresentation
      • SingleCompGCNRepresentation
    • SubsetRepresentation
      • SubsetRepresentation
    • CombinedRepresentation
      • CombinedRepresentation
        • CombinedRepresentation.combine()
    • TensorTrainRepresentation
      • TensorTrainRepresentation
        • TensorTrainRepresentation.check_assignment()
        • TensorTrainRepresentation.check_factors()
        • TensorTrainRepresentation.create_default_assignment()
        • TensorTrainRepresentation.factor_sizes()
        • TensorTrainRepresentation.get_shapes_and_einsum_eq()
        • TensorTrainRepresentation.iter_extra_repr()
    • TextRepresentation
      • TextRepresentation
        • TextRepresentation.from_dataset()
        • TextRepresentation.from_triples_factory()
    • TransformedRepresentation
      • TransformedRepresentation
    • WikidataTextRepresentation
      • WikidataTextRepresentation
  • Class Inheritance Diagram
• Initialization
  • Functions
    • xavier_uniform
      • xavier_uniform_()
    • xavier_normal
      • xavier_normal_()
    • init_phases
      • init_phases()
  • Classes
    • PretrainedInitializer
      • PretrainedInitializer
        • PretrainedInitializer.__call__()
        • PretrainedInitializer.as_embedding()
    • LabelBasedInitializer
      • LabelBasedInitializer
        • LabelBasedInitializer.from_triples_factory()
    • WeisfeilerLehmanInitializer
      • WeisfeilerLehmanInitializer
    • RandomWalkPositionalEncodingInitializer
      • RandomWalkPositionalEncodingInitializer
  • Class Inheritance Diagram
• Message Passing
  • Classes
    • RGCNRepresentation
      • RGCNRepresentation
        • RGCNRepresentation.post_parameter_update()
        • RGCNRepresentation.reset_parameters()
    • Decomposition
      • Decomposition
        • Decomposition.forward()
        • Decomposition.forward_horizontally_stacked()
        • Decomposition.forward_vertically_stacked()
        • Decomposition.iter_extra_repr()
        • Decomposition.reset_parameters()
    • BasesDecomposition
      • BasesDecomposition
        • BasesDecomposition.base_weights
        • BasesDecomposition.bases
        • BasesDecomposition.forward_horizontally_stacked()
        • BasesDecomposition.forward_vertically_stacked()
        • BasesDecomposition.iter_extra_repr()
        • BasesDecomposition.reset_parameters()
    • BlockDecomposition
      • BlockDecomposition
        • BlockDecomposition.forward_horizontally_stacked()
        • BlockDecomposition.forward_vertically_stacked()
        • BlockDecomposition.iter_extra_repr()
        • BlockDecomposition.reset_parameters()
  • Class Inheritance Diagram
• PyG Message Passing
  • Classes
    • MessagePassingRepresentation
      • MessagePassingRepresentation
        • MessagePassingRepresentation.pass_messages()
    • SimpleMessagePassingRepresentation
      • SimpleMessagePassingRepresentation
        • SimpleMessagePassingRepresentation.pass_messages()
    • FeaturizedMessagePassingRepresentation
      • FeaturizedMessagePassingRepresentation
        • FeaturizedMessagePassingRepresentation.pass_messages()
    • TypedMessagePassingRepresentation
      • TypedMessagePassingRepresentation
        • TypedMessagePassingRepresentation.pass_messages()
  • Class Inheritance Diagram
• Weighting
  • Classes
    • EdgeWeighting
      • EdgeWeighting
        • EdgeWeighting.needs_message
        • EdgeWeighting.forward()
    • InverseInDegreeEdgeWeighting
      • InverseInDegreeEdgeWeighting
        • InverseInDegreeEdgeWeighting.forward()
    • InverseOutDegreeEdgeWeighting
      • InverseOutDegreeEdgeWeighting
        • InverseOutDegreeEdgeWeighting.forward()
    • SymmetricEdgeWeighting
      • SymmetricEdgeWeighting
        • SymmetricEdgeWeighting.forward()
    • AttentionEdgeWeighting
      • AttentionEdgeWeighting
        • AttentionEdgeWeighting.needs_message
        • AttentionEdgeWeighting.forward()
  • Class Inheritance Diagram
• Combinations
  • Classes
    • Combination
      • Combination
        • Combination.forward()
        • Combination.output_shape()
    • ComplexSeparatedCombination
      • ComplexSeparatedCombination
        • ComplexSeparatedCombination.forward()
        • ComplexSeparatedCombination.output_shape()
    • ConcatCombination
      • ConcatCombination
        • ConcatCombination.forward()
        • ConcatCombination.iter_extra_repr()
    • ConcatAggregationCombination
      • ConcatAggregationCombination
        • ConcatAggregationCombination.forward()
        • ConcatAggregationCombination.iter_extra_repr()
    • ConcatProjectionCombination
      • ConcatProjectionCombination
        • ConcatProjectionCombination.forward()
    • GatedCombination
      • GatedCombination
        • GatedCombination.forward()
  • Class Inheritance Diagram
• Perceptron
  • ConcatMLP
    • ConcatMLP.forward()
• Utilities
  • ShapeError
    • ShapeError.verify()
  • WikidataCache
    • WikidataCache.WIKIDATA_ENDPOINT
    • WikidataCache.get_descriptions()
    • WikidataCache.get_image_paths()
    • WikidataCache.get_labels()
    • WikidataCache.query()
    • WikidataCache.query_text()
    • WikidataCache.verify_ids()
  • adjacency_tensor_to_stacked_matrix()
  • safe_diagonal()
  • use_horizontal_stacking()
• NodePiece
  • pykeen.nn.node_piece Package
    • Classes
      • AnchorSearcher
        • AnchorSearcher
          • AnchorSearcher.__call__()
      • ScipySparseAnchorSearcher
        • ScipySparseAnchorSearcher
          • ScipySparseAnchorSearcher.__call__()
          • ScipySparseAnchorSearcher.bfs()
          • ScipySparseAnchorSearcher.create_adjacency()
          • ScipySparseAnchorSearcher.iter_extra_repr()
          • ScipySparseAnchorSearcher.select()
      • SparseBFSSearcher
        • SparseBFSSearcher
          • SparseBFSSearcher.__call__()
          • SparseBFSSearcher.bfs()
          • SparseBFSSearcher.create_adjacency()
          • SparseBFSSearcher.iter_extra_repr()
          • SparseBFSSearcher.select()
      • CSGraphAnchorSearcher
        • CSGraphAnchorSearcher
          • CSGraphAnchorSearcher.__call__()
      • PersonalizedPageRankAnchorSearcher
        • PersonalizedPageRankAnchorSearcher
          • PersonalizedPageRankAnchorSearcher.__call__()
          • PersonalizedPageRankAnchorSearcher.iter_extra_repr()
          • PersonalizedPageRankAnchorSearcher.precalculate_anchor_ppr()
      • AnchorSelection
        • AnchorSelection
          • AnchorSelection.__call__()
          • AnchorSelection.filter_unique()
          • AnchorSelection.iter_extra_repr()
      • SingleSelection
        • SingleSelection
          • SingleSelection.__call__()
          • SingleSelection.rank()
      • DegreeAnchorSelection
        • DegreeAnchorSelection
          • DegreeAnchorSelection.rank()
      • MixtureAnchorSelection
        • MixtureAnchorSelection
          • MixtureAnchorSelection.__call__()
          • MixtureAnchorSelection.iter_extra_repr()
      • PageRankAnchorSelection
        • PageRankAnchorSelection
          • PageRankAnchorSelection.iter_extra_repr()
          • PageRankAnchorSelection.rank()
      • RandomAnchorSelection
        • RandomAnchorSelection
          • RandomAnchorSelection.rank()
      • Tokenizer
        • Tokenizer
          • Tokenizer.__call__()
      • RelationTokenizer
        • RelationTokenizer
          • RelationTokenizer.__call__()
      • AnchorTokenizer
        • AnchorTokenizer
          • AnchorTokenizer.__call__()
      • MetisAnchorTokenizer
        • MetisAnchorTokenizer
          • MetisAnchorTokenizer.__call__()
      • PrecomputedPoolTokenizer
        • PrecomputedPoolTokenizer
          • PrecomputedPoolTokenizer.__call__()
      • PrecomputedTokenizerLoader
        • PrecomputedTokenizerLoader
          • PrecomputedTokenizerLoader.__call__()
      • GalkinPrecomputedTokenizerLoader
        • GalkinPrecomputedTokenizerLoader
          • GalkinPrecomputedTokenizerLoader.__call__()
      • TorchPrecomputedTokenizerLoader
        • TorchPrecomputedTokenizerLoader
          • TorchPrecomputedTokenizerLoader.__call__()
          • TorchPrecomputedTokenizerLoader.save()
      • TokenizationRepresentation
        • TokenizationRepresentation
          • TokenizationRepresentation.num_tokens
          • TokenizationRepresentation.from_tokenizer()
          • TokenizationRepresentation.iter_extra_repr()
          • TokenizationRepresentation.save_assignment()
      • NodePieceRepresentation
        • NodePieceRepresentation
          • NodePieceRepresentation.estimate_diversity()
      • HashDiversityInfo
        • HashDiversityInfo
    • Class Inheritance Diagram