diff options
Diffstat (limited to 'abml/learn_dist.py')
-rw-r--r-- | abml/learn_dist.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/abml/learn_dist.py b/abml/learn_dist.py new file mode 100644 index 0000000..58e4968 --- /dev/null +++ b/abml/learn_dist.py @@ -0,0 +1,15 @@ +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('Name', type=str, help='Predicate name.') +args = parser.parse_args() +name = args.Name + +data = Table('data/{}/programs-train'.format(name)) + +rule_learner = rp.create_learner(name, evds=False) +rule_learner.calculate_evds(data) +pickle.dump(rule_learner.evds, open("data/{}/evds.pickle".format(name), "wb")) |