In [3]:

    

import sys
sys.path.append('/home/wrwt/Programming/pygraphmodels')


    

In [4]:

    

import graphmodels as gm
import numpy as np
import pandas as pd


    

In [26]:

    

a = np.random.randint(2, size=1000)
b = np.random.randint(2, size=1000)
c = a + b
data = np.transpose(np.vstack([a, b, c]))
df = pd.DataFrame(data=data, columns=['a', 'b', 'c'])


    

In [27]:

    

tf = gm.TableFactor(['a', 'b', 'c'], ['a', 'b', 'c'])


    

In [28]:

    

tf.fit(df)


    

    
Out[28]:




a
b
c
P
0
0
0
0.259
0
0
1
0.000
0
0
2
0.000
0
1
0
0.000
0
1
1
0.251
0
1
2
0.000
1
0
0
0.000
1
0
1
0.245
1
0
2
0.000
1
1
0
0.000
1
1
1
0.000
1
1
2
0.245

In [29]:

    

tf.marginalize('a', 'b')


    

    
Out[29]:




c
P
0
0.259
1
0.496
2
0.245

In [ ]: