Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix deep learning tutorial

This commit is contained in:
Matthew Honnibal 2016-10-20 03:21:56 +02:00
parent a660585b12
commit d17546681c
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
examples/deep_learning_keras.py
View File

@ -4,9 +4,8 @@ import random
import cytoolz import cytoolz
import numpy import numpy
from keras.layers import Sequential, LSTM, Dense, Embedding, Dropout from keras.models import Sequential, model_from_json
from keras.wrappers import Bidirectional from keras.layers import LSTM, Dense, Embedding, Dropout, Bidirectional
from keras import model_from_json
import cPickle as pickle import cPickle as pickle
import spacy import spacy
@ -127,17 +126,16 @@ def read_data(data_dir, limit=0):
@plac.annotations( @plac.annotations(
language=("The language to train", "positional", None, str, ['en','de', 'zh']), train_dir=("Location of training file or directory"),
train_loc=("Location of training file or directory"), dev_dir=("Location of development file or directory"),
dev_loc=("Location of development file or directory"),
model_dir=("Location of output model directory",), model_dir=("Location of output model directory",),
is_runtime=("Demonstrate run-time usage", "flag", "r", bool), is_runtime=("Demonstrate run-time usage", "flag", "r", bool),
nr_hidden=("Number of hidden units", "flag", "H", int), nr_hidden=("Number of hidden units", "option", "H", int),
max_length=("Maximum sentence length", "flag", "L", int), max_length=("Maximum sentence length", "option", "L", int),
dropout=("Dropout", "flag", "d", float), dropout=("Dropout", "option", "d", float),
nr_epoch=("Number of training epochs", "flag", "i", int), nb_epoch=("Number of training epochs", "option", "i", int),
batch_size=("Size of minibatches for training LSTM", "flag", "b", int), batch_size=("Size of minibatches for training LSTM", "option", "b", int),
nr_examples=("Limit to N examples", "flag", "n", int) nr_examples=("Limit to N examples", "option", "n", int)
) )
def main(model_dir, train_dir, dev_dir, def main(model_dir, train_dir, dev_dir,
is_runtime=False, is_runtime=False,