def load_shakespeare(path): """ Deprecated, moved to neon.data.dataloaders. """ logger.error('This function has moved, import from neon.data.dataloaders') from neon.data.dataloaders import load_shakespeare # noqa return load_shakespeare(path)
# parse the command line arguments default_overrides = dict(save_path='rnn_text_gen.pickle', serialize=1, batch_size=64) parser = NeonArgparser(__doc__, default_overrides=default_overrides) args = parser.parse_args() # hyperparameters time_steps = 64 hidden_size = 512 gradient_clip_value = 5 # download shakespeare text data_path = load_shakespeare(path=args.data_dir) train_path, valid_path = Text.create_valid_file(data_path) # load data and parse on character-level train_set = Text(time_steps, train_path) valid_set = Text(time_steps, valid_path, vocab=train_set.vocab) # weight initialization init = Uniform(low=-0.08, high=0.08) # model initialization layers = [ LSTM(hidden_size, init, activation=Logistic(), gate_activation=Tanh()), Affine(len(train_set.vocab), init, bias=init, activation=Softmax()) ] model = Model(layers=layers)
args.callback_args['save_path'] = args.save_path if args.callback_args['serialize'] is None: args.callback_args['serialize'] = 1 # hyperparameters args.batch_size = 64 time_steps = 64 hidden_size = 512 gradient_clip_value = 5 # setup backend be = gen_backend(**extract_valid_args(args, gen_backend)) # download shakespeare text data_path = load_shakespeare(path=args.data_dir) train_path, valid_path = Text.create_valid_file(data_path) # load data and parse on character-level train_set = Text(time_steps, train_path) valid_set = Text(time_steps, valid_path, vocab=train_set.vocab) # weight initialization init = Uniform(low=-0.08, high=0.08) # model initialization layers = [ LSTM(hidden_size, init, activation=Logistic(), gate_activation=Tanh()), Affine(len(train_set.vocab), init, bias=init, activation=Softmax()) ] model = Model(layers=layers)
def load_shakespeare(path): logger.error('This function has moved, import from neon.data.dataloaders') from neon.data.dataloaders import load_shakespeare # noqa return load_shakespeare(path)