diff options
Diffstat (limited to 'abml/learn_dist.py')
| -rw-r--r-- | abml/learn_dist.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/abml/learn_dist.py b/abml/learn_dist.py index 58e4968..44e19c3 100644 --- a/abml/learn_dist.py +++ b/abml/learn_dist.py @@ -1,15 +1,16 @@ +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('Name', type=str, help='Predicate name.') +parser.add_argument('path', help='path to data directory') args = parser.parse_args() -name = args.Name +path = args.path -data = Table('data/{}/programs-train'.format(name)) +data = Table(os.path.join(path, 'programs-train')) -rule_learner = rp.create_learner(name, evds=False) +rule_learner = rp.create_learner(path, evds=False) rule_learner.calculate_evds(data) -pickle.dump(rule_learner.evds, open("data/{}/evds.pickle".format(name), "wb")) +pickle.dump(rule_learner.evds, open(os.path.join(path, 'evds.pickle'), "wb")) |