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