Python ConstraintBasedEstimator.edges Examples

Example #1

 def test_estimate(self):
     data = pd.DataFrame(np.random.randint(0, 3, size=(1000, 3)),
                         columns=list('XYZ'))
     data['sum'] = data.sum(axis=1)
     model = ConstraintBasedEstimator(data).estimate()
     self.assertSetEqual(set(model.edges()),
                         set([('Z', 'sum'), ('X', 'sum'), ('Y', 'sum')]))

Example #2

File: test_ConstraintBasedEstimator.py Project: sajib-4414/comparative-study-with-pgmpy

 def test_estimate(self):
     data = pd.DataFrame(np.random.randint(0, 3, size=(1000, 3)),
                         columns=list("XYZ"))
     data["sum"] = data.sum(axis=1)
     model = ConstraintBasedEstimator(data).estimate()
     self.assertSetEqual(set(model.edges()),
                         set([("Z", "sum"), ("X", "sum"), ("Y", "sum")]))

Example #3

File: test_ConstraintBasedEstimator.py Project: MariosRichards/BES_analysis_code

 def test_estimate(self):
     data = pd.DataFrame(np.random.randint(0, 3, size=(1000, 3)), columns=list('XYZ'))
     data['sum'] = data.sum(axis=1)
     model = ConstraintBasedEstimator(data).estimate()
     self.assertSetEqual(set(model.edges()),
                         set([('Z', 'sum'), ('X', 'sum'), ('Y', 'sum')]))