import os.path
import pickle
import argparse
from Orange.data import Table
import abml.rules_prolog as rp

parser = argparse.ArgumentParser(description='Learn and test rules for prolog programs.')
parser.add_argument('path', help='path to data directory')
args = parser.parse_args()
path = args.path

data = Table(os.path.join(path, 'programs-train'))

rule_learner = rp.create_learner(path, evds=False)
rule_learner.calculate_evds(data)
pickle.dump(rule_learner.evds, open(os.path.join(path, 'evds.pickle'), "wb"))