def test_dfmc(self): rnds = np.random.RandomState(0) R12 = rnds.rand(30, 30) R13 = rnds.rand(30, 30) t1 = ObjectType('type1', 50) t2 = ObjectType('type2', 30) t3 = ObjectType('type3', 10) fusion_graph = FusionGraph() relations = [Relation(R12, t1, t2), Relation(R13, t1, t3)] fusion_graph.add_relations_from(relations) fuser = Dfmc(init_type='random', random_state=rnds, n_run=3 ).fuse(fusion_graph) self.assertEqual(len(list(fuser.factor(t1))), 3) self.assertEqual(len(list(fuser.factor(t2))), 3) self.assertEqual(len(list(fuser.factor(t3))), 3) self.assertEqual(len(list(fuser.backbone(relations[0]))), 3) self.assertEqual(len(list(fuser.backbone(relations[1]))), 3) for object_type in [t1, t2, t3]: for factor in fuser.factor(object_type): self.assertEqual(factor.shape, (30, object_type.rank)) G1 = fuser.factor(t1, run=1) S13 = fuser.backbone(relations[1], run=1) G3 = fuser.factor(t3, run=1) R13_hat = np.dot(G1, np.dot(S13, G3.T)) completed = fuser.complete(relations[1], run=1) np.testing.assert_almost_equal(completed, R13_hat)
def test_dfmc(self): rnds = np.random.RandomState(0) R12 = rnds.rand(30, 30) R13 = rnds.rand(30, 30) t1 = ObjectType('type1', 50) t2 = ObjectType('type2', 30) t3 = ObjectType('type3', 10) fusion_graph = FusionGraph() relations = [Relation(R12, t1, t2), Relation(R13, t1, t3)] fusion_graph.add_relations_from(relations) fuser = Dfmc(init_type='random', random_state=rnds, n_run=3).fuse(fusion_graph) self.assertEqual(len(list(fuser.factor(t1))), 3) self.assertEqual(len(list(fuser.factor(t2))), 3) self.assertEqual(len(list(fuser.factor(t3))), 3) self.assertEqual(len(list(fuser.backbone(relations[0]))), 3) self.assertEqual(len(list(fuser.backbone(relations[1]))), 3) for object_type in [t1, t2, t3]: for factor in fuser.factor(object_type): self.assertEqual(factor.shape, (30, object_type.rank)) G1 = fuser.factor(t1, run=1) S13 = fuser.backbone(relations[1], run=1) G3 = fuser.factor(t3, run=1) R13_hat = np.dot(G1, np.dot(S13, G3.T)) completed = fuser.complete(relations[1], run=1) np.testing.assert_almost_equal(completed, R13_hat)
def test_dfmc(self): rnds = np.random.RandomState(0) R12_1 = np.random.rand(30, 30) R12_2 = np.random.rand(30, 30) R13 = np.random.rand(30, 20) t1 = ObjectType('type1', 30) t2 = ObjectType('type2', 30) t3 = ObjectType('type3', 20) relations = [ Relation(R12_1, t1, t2), Relation(R12_2, t1, t2), Relation(R13, t1, t3) ] fusion_graph = FusionGraph() fusion_graph.add_relations_from(relations) self.assertEqual(len(fusion_graph.relations), 3) self.assertEqual(len(fusion_graph.object_types), 3) fuser = Dfmc(init_type='random', random_state=rnds).fuse(fusion_graph) self.assertEqual(fuser.backbone(relations[0]).shape, (30, 30)) self.assertEqual(fuser.backbone(relations[1]).shape, (30, 30)) self.assertEqual(fuser.backbone(relations[2]).shape, (30, 20)) G1 = fuser.factor(t1) G2 = fuser.factor(t2) S12_1 = fuser.backbone(relations[0]) S12_2 = fuser.backbone(relations[1]) R12_1_hat = np.dot(G1, np.dot(S12_1, G2.T)) R12_2_hat = np.dot(G1, np.dot(S12_2, G2.T)) np.testing.assert_almost_equal(fuser.complete(relations[0]), R12_1_hat) np.testing.assert_almost_equal(fuser.complete(relations[1]), R12_2_hat)
def test_dfmc(self): rnds = np.random.RandomState(0) R12_1 = np.random.rand(30, 30) R12_2 = np.random.rand(30, 30) R13 = np.random.rand(30, 20) t1 = ObjectType('type1', 30) t2 = ObjectType('type2', 30) t3 = ObjectType('type3', 20) relations = [Relation(R12_1, t1, t2), Relation(R12_2, t1, t2), Relation(R13, t1, t3)] fusion_graph = FusionGraph() fusion_graph.add_relations_from(relations) self.assertEqual(len(fusion_graph.relations), 3) self.assertEqual(len(fusion_graph.object_types), 3) fuser = Dfmc(init_type='random', random_state=rnds).fuse(fusion_graph) self.assertEqual(fuser.backbone(relations[0]).shape, (30, 30)) self.assertEqual(fuser.backbone(relations[1]).shape, (30, 30)) self.assertEqual(fuser.backbone(relations[2]).shape, (30, 20)) G1 = fuser.factor(t1) G2 = fuser.factor(t2) S12_1 = fuser.backbone(relations[0]) S12_2 = fuser.backbone(relations[1]) R12_1_hat = np.dot(G1, np.dot(S12_1, G2.T)) R12_2_hat = np.dot(G1, np.dot(S12_2, G2.T)) np.testing.assert_almost_equal(fuser.complete(relations[0]), R12_1_hat) np.testing.assert_almost_equal(fuser.complete(relations[1]), R12_2_hat)
def test_dfmc(self): rnds = np.random.RandomState(0) R12 = rnds.rand(50, 30) t1 = ObjectType('type1', 50) t2 = ObjectType('type2', 30) relation = Relation(R12, t1, t2) fusion_graph = FusionGraph() fusion_graph.add_relation(relation) fuser = Dfmc(init_type='random', random_state=rnds).fuse(fusion_graph) self.assertEqual(fuser.backbone(relation).shape, (50, 30)) self.assertEqual(fuser.factor(t1).shape, (50, 50)) self.assertEqual(fuser.factor(t2).shape, (30, 30)) np.testing.assert_almost_equal(fuser.complete(relation), relation.data)
def test_postprocessors(self): rnds = np.random.RandomState(0) R12 = rnds.rand(50, 30) R12 = np.ma.masked_greater(R12, 0.7) t1 = ObjectType('type1', 50) t2 = ObjectType('type2', 30) def postprocessor(data): return data - 10 relation = Relation(R12, t1, t2, name='R', postprocessor=postprocessor) fusion_graph = FusionGraph() fusion_graph.add_relation(relation) fuser = Dfmc(init_type='random', random_state=rnds).fuse(fusion_graph) self.assertEqual(fuser.backbone(relation).shape, (50, 30)) self.assertEqual(fuser.factor(t1).shape, (50, 50)) self.assertEqual(fuser.factor(t2).shape, (30, 30)) trnf = relation.data - 10 np.testing.assert_almost_equal(fuser.complete(relation), trnf) np.testing.assert_equal(fusion_graph.get_relation('R').data, R12)