def main():
args = get_args()
cfg = Config()
keys = [
["vectorizer", "batch_size"],
["vectorizer", "samples"],
["vectorizer", "vector_size"],
["vectorizer", "epochs"],
]
cfg.setup(keys)
v = Vectorize(
indir=args.indir,
batch_size=cfg.batch_size,
n_samples=cfg.samples,
epochs=cfg.epochs,
vector_size=cfg.vector_size,
)
v.run()
def get_config():
cfg = Config()
keys = [
["seq2vec", "gpu"],
["seq2vec", "load"],
["seq2vec", "epochs"],
["seq2vec", "lr"],
["seq2vec", "weight_decay"],
["seq2vec", "hidden_size"],
["seq2vec", "n_layers"],
["seq2vec", "bidirectional"],
["seq2vec", "batch_size"],
["seq2vec", "max_seqlen"],
["seq2vec", "train_samples"],
["seq2vec", "model_dir"],
["seq2vec", "predict_intervals"],
["seq2vec", "predict_samples"],
["visdom", "server"],
["visdom", "port"],
]
cfg.setup(keys)
return cfg
