def run(args=None):
usage = "usage : %prog [options]"
parser = optparse.OptionParser(usage=usage)
parser.add_option("--test", action="store_true", dest="test", default=False)
parser.add_option("--output_dim", dest="output_dim", type="int", default=0)
parser.add_option("--out_file", dest="out_file", type="string",
default="models/baseline.bin")
parser.add_option("--in_file", dest="in_file", type="string",
default="models/baseline.bin")
parser.add_option("--data", dest="data", type="string", default="train")
(opts, args) = parser.parse_args(args)
# Testing
if opts.test:
test(opts.in_file, opts.data)
return
print("Loading data...")
# load training data
trees = load_trees()
if opts.output_dim == 0:
opts.output_dim = len(load_label_map())
baseline = Baseline(opts.output_dim)
baseline.train(trees)
with open(opts.out_file, 'wb') as fid:
pickle.dump(opts, fid)
baseline.to_file(fid)
def run(args=None):
usage = "usage : %prog [options]"
parser = optparse.OptionParser(usage=usage)
parser.add_option("--test",
action="store_true",
dest="test",
default=False)
parser.add_option("--output_dim", dest="output_dim", type="int", default=0)
parser.add_option("--out_file",
dest="out_file",
type="string",
default="models/baseline.bin")
parser.add_option("--in_file",
dest="in_file",
type="string",
default="models/baseline.bin")
parser.add_option("--data", dest="data", type="string", default="train")
(opts, args) = parser.parse_args(args)
# Testing
if opts.test:
test(opts.in_file, opts.data)
return
print("Loading data...")
# load training data
trees = load_trees()
if opts.output_dim == 0:
opts.output_dim = len(load_label_map())
baseline = Baseline(opts.output_dim)
baseline.train(trees)
with open(opts.out_file, 'wb') as fid:
pickle.dump(opts, fid)
baseline.to_file(fid)
