def setUp(self) -> None:
"""Prepare the training, testing, and validation triples factories."""
self.base_tf = generate_triples_factory(
num_entities=50,
num_relations=9,
num_triples=500,
)
self.training, self.testing, self.validation = self.base_tf.split([0.8, 0.1, 0.1])
def _pre_instantiation_hook(
self, kwargs: MutableMapping[str, Any]
) -> MutableMapping[str, Any]: # noqa: D102
kwargs = super()._pre_instantiation_hook(kwargs=kwargs)
kwargs["triples_factory"] = generate_triples_factory(
num_entities=self.num,
num_relations=self.num_relations,
num_triples=self.num_triples,
)
return kwargs
def test_generate_triples_factory(self):
"""Test generating a triples factory."""
for random_state in range(100):
tf = generate_triples_factory(
num_entities=self.num_entities,
num_relations=self.num_relations,
num_triples=self.num_triples,
random_state=random_state,
)
self.assertEqual(self.num_triples, tf.mapped_triples.shape[0])
self.assert_consecutive(get_entities(tf.mapped_triples))
self.assert_consecutive(get_relations(tf.mapped_triples))
def _pre_instantiation_hook(
self, kwargs: MutableMapping[str, Any]
) -> MutableMapping[str, Any]: # noqa: D102
kwargs = super()._pre_instantiation_hook(kwargs=kwargs)
kwargs["combined"] = pykeen.nn.emb.CombinedCompGCNRepresentations(
triples_factory=generate_triples_factory(
num_entities=self.num_entities,
num_relations=self.num_relations,
num_triples=self.num_triples,
create_inverse_triples=True,
),
embedding_specification=EmbeddingSpecification(
embedding_dim=self.dim),
dims=self.dim,
)
return kwargs