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);
}
