def test_ermlp_predict(self):
"""Test ERMLP's predict function."""
ermlp = ERMLP(config=ERMLP_CONFIG)
predictions = ermlp.predict(triples=TEST_TRIPLES)
self.assertEqual(len(predictions), len(TEST_TRIPLES))
self.assertTrue(type(predictions.shape[0]), float)
def test_ermlp_predict(self):
"""Test ERMLP's predict function."""
ermlp = ERMLP(**ERMLP_CONFIG)
ermlp.num_entities = ERMLP_CONFIG[NUM_ENTITIES]
ermlp.num_relations = ERMLP_CONFIG[NUM_RELATIONS]
predictions = ermlp.predict(triples=TEST_TRIPLES)
self.assertEqual(len(predictions), len(TEST_TRIPLES))
self.assertTrue(type(predictions.shape[0]), float)