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