a6

#program 6
# import numpy as np
# import csv
import pandas as pd
from pgmpy.models import BayesianNetwork
from pgmpy.estimators import MaximumLikelihoodEstimator
from pgmpy.inference import VariableElimination
heart=pd.read_csv('P6Data.csv')
heart=heart.replace('?',"null")
print(heart.head())
model=BayesianNetwork([('age','target'),('sex','target'),
('cp','target'),('exang','target'),('target','restecg'),
('target','chol')])
model.fit(heart,estimator=MaximumLikelihoodEstimator)
heart_infer=VariableElimination(model)
q1=heart_infer.query(variables=['target'],evidence={'age':50})
print(q1)
q2=heart_infer.query(variables=['target'],evidence={'chol':300})
print(q2)