def test_get_class_data(self): kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv') kb.add_data('..\\..\\test_data\\Test_Data_C.csv') print(kb.get_data()) print() print(kb.get_class_data(['A', 'C'])) print() print(kb.get_class_data(['B'])) print(kb.get_class_data(['Z'])) print(kb.get_class_data(['A', 'B', 'Z']))
def test_add_data(self): kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv') print(kb.get_data()) print() kb.add_data('..\\..\\test_data\\Test_Data_C.csv') print(kb.get_data())
def test_update_data(self): kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv') print(kb.get_data()) print() kb.update_data('..\\..\\test_data\\test_data_A_update.csv') print(kb.get_data())
from bayesian_network import BayesianNetwork from knowledge_base import KnowledgeBase from graph_builder import GraphBuilder if __name__ == "__main__": _data_file_path = '..\\ExcelFiles\\courses_and_grades.csv' _relations_file_path = '..\\..\\Data\\combined_course_structure.csv' knowledge_base = KnowledgeBase(_relations_file_path, _data_file_path) builder = GraphBuilder() builder = builder.build_nodes(list(knowledge_base.get_data().columns)) builder = builder.add_parents(knowledge_base.get_relations()) builder = builder.add_children() builder = builder.build_edges() graph = builder.build_graph() nodes = graph.get_nodes() # bayes_net = BayesianNetwork(knowledge_base, graph) # print(bayes_net.get_graph().get_node('Calculus and Analytic Geometry I').get_parents())