void run_char_rnn(int argc, char **argv) { if (argc < 4) { fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]); return; } char *filename = find_char_arg(argc, argv, "-file", "data/shakespeare.txt"); char *seed = find_char_arg(argc, argv, "-seed", "\n\n"); int len = find_int_arg(argc, argv, "-len", 1000); real_t temp = find_real_t_arg(argc, argv, "-temp", .7); int rseed = find_int_arg(argc, argv, "-srand", time(0)); int clear = find_arg(argc, argv, "-clear"); int tokenized = find_arg(argc, argv, "-tokenized"); char *tokens = find_char_arg(argc, argv, "-tokens", 0); char *cfg = argv[3]; char *weights = (argc > 4) ? argv[4] : 0; if (0 == strcmp(argv[2], "train")) train_char_rnn(cfg, weights, filename, clear, tokenized); else if (0 == strcmp(argv[2], "valid")) valid_char_rnn(cfg, weights, seed); else if (0 == strcmp(argv[2], "validtactic")) valid_tactic_rnn(cfg, weights, seed); else if (0 == strcmp(argv[2], "vec")) vec_char_rnn(cfg, weights, seed); else if (0 == strcmp(argv[2], "generate")) test_char_rnn(cfg, weights, len, seed, temp, rseed, tokens); else if (0 == strcmp(argv[2], "generatetactic")) test_tactic_rnn(cfg, weights, len, temp, rseed, tokens); }
void run_char_rnn(int argc, char **argv) { if(argc < 4){ fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]); return; } char *filename = find_char_arg(argc, argv, "-file", "data/shakespeare.txt"); char *seed = find_char_arg(argc, argv, "-seed", "\n"); int len = find_int_arg(argc, argv, "-len", 100); float temp = find_float_arg(argc, argv, "-temp", 1); int rseed = find_int_arg(argc, argv, "-srand", time(0)); char *cfg = argv[3]; char *weights = (argc > 4) ? argv[4] : 0; if(0==strcmp(argv[2], "train")) train_char_rnn(cfg, weights, filename); else if(0==strcmp(argv[2], "test")) test_char_rnn(cfg, weights, len, seed, temp, rseed); }