From cc41068e6b4566633ebc7a841290ce167671d2bf Mon Sep 17 00:00:00 2001 From: Timotej Lazar Date: Tue, 18 Apr 2017 17:25:18 +0200 Subject: Store student submissions in files instead of .pickle --- abml/learn_dist.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'abml/learn_dist.py') 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")) -- cgit v1.2.1