def test_predict(self):
lines = sys.stdin.readlines()
reader = IOBReader(lines)
extractors = [
FieldExtractor(reader.getPosition('FORM')),
FieldExtractor(reader.getPosition('POS')),
CapitalExtractor(reader.getPosition('FORM')),
]
params = {'epochs':25, 'learning_rate':0.01, 'window_size':3, 'name_model':'model.ckpt'}
classifier = Classifier(reader, extractors, LinearEstimator, **params)
predicted = classifier.predict()
#self.assertEqual(len(dataset), len(predicted))
labels_idx_rev = {v:k for k,v in reader.vocabulary[reader.getPosition('LABEL')].items()}
i = 0
for line in lines:
line = line.strip()
if line:
print '%s\t%s\t%s' % (line.split()[0], line.split()[1], labels_idx_rev[predicted[i]])
i += 1
else:
print
def test_predict(self):
lines = sys.stdin.readlines()
reader = IOBReader(lines)
extractors = []
params = {
'epochs': 100,
'learning_rate': 0.01,
'window_size': 5,
'name_model': 'model_we.ckpt',
'word_embeddings_file': 'data/vectors.txt'
}
classifier = WordEmbeddingsClassifier(reader, extractors, WordEmbeddingsEstimator, **params)
predicted = classifier.predict()
labels_idx_rev = {v:k for k,v in reader.vocabulary[reader.getPosition('LABEL')].items()}
i = 0
for line in lines:
line = line.strip()
if line:
print '%s\t%s\t%s' % (line.split()[0], line.split()[1], labels_idx_rev[predicted[i]])
i += 1
else:
print
def test_reader(self):
filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'test_reader.iob')
reader = IOBReader(open(filename))
reader = IOBReader(sys.stdin)
sentences, labels = reader.read()
self.assertEqual(len(sentences), len(labels))
def test_train(self):
f = {
'fields': [
{'position': 0, 'name': 'FORM', 'type': str},
{'position': 1, 'name': 'POS', 'type': str},
{'position': 2, 'name': 'LABEL', 'type': str}
]
}
reader = IOBReader(sys.stdin, separator='\t', format=f)
extractors = [
FieldExtractor(reader.getPosition('FORM')),
FieldExtractor(reader.getPosition('POS')),
CapitalExtractor(reader.getPosition('FORM')),
]
params = {'epochs':25, 'learning_rate':0.01, 'window_size':3, 'name_model':'model.ckpt'}
classifier = Classifier(reader, extractors, LinearEstimator, **params)
classifier.train()
def main():
parser = argparse.ArgumentParser(description='Named Entity Recognition with TensorFlow')
subparsers = parser.add_subparsers()
parser_train = subparsers.add_parser('train')
parser_train.set_defaults(which='train')
parser_train.add_argument('-e', '--epochs', help='epochs number', type=int, required=True)
parser_train.add_argument('-l', '--learning-rate', help='learning rate', type=float, required=True)
parser_train.add_argument('-o', '--optimizer', help='optimizer', type=str, required=True, choices=OPTIMIZERS.keys())
parser_tag = subparsers.add_parser('tag')
parser_tag.set_defaults(which='tag')
# common arguments
for p in (parser_train, parser_tag):
p.add_argument('-m', '--model', help='model-file', type=str, required=True)
p.add_argument('-r', '--reader-file', help='reader file', type=str, required=True)
p.add_argument('-w', '--word-embeddings', help='word embeddings', type=str, required=False)
p.add_argument('-et', '--word-embeddings-type', help='word embeddings type', type=str, required=False)
p.add_argument('-i', '--input-file', help='input file', type=str, required=False)
p.add_argument('-t', '--type', help='estimator type', type=str, required=True, choices=ESTIMATORS.keys())
p.add_argument('-wi', '--window', help='context window size', type=int, required=True)
p.add_argument('-nl', '--num-layers', help='number layers for multi rnn estimator', type=int, required=False)
p.add_argument('-f', '--feats-conf', help='add the feats in the conf number', type=int, required=False)
args = parser.parse_args()
infile = args.input_file if args.input_file is not None else sys.stdin
if args.which == 'train':
reader = IOBReader(infile, separator='\t')
extractors = []
if args.feats_conf is not None and args.feats_conf != 0:
extractors = [
FieldExtractor(reader.getPosition('FORM')),
FieldExtractor(reader.getPosition('POS')),
CapitalExtractor(reader.getPosition('FORM'))
]
params = {
'epochs': args.epochs,
'learning_rate': args.learning_rate,
'window_size': args.window,
'name_model': args.model,
'word_embeddings_file': args.word_embeddings,
'reader_file': args.reader_file,
'optimizer': OPTIMIZERS[args.optimizer],
'num_layers': args.num_layers
}
classifier = WordEmbeddingsClassifier(reader, extractors, ESTIMATORS[args.type], **params)
classifier.train()
elif args.which == 'tag':
lines = sys.stdin.readlines()
reader = IOBReader(lines)
extractors = []
if args.feats_conf is not None and args.feats_conf != 0:
extractors = [
FieldExtractor(reader.getPosition('FORM')),
FieldExtractor(reader.getPosition('POS')),
CapitalExtractor(reader.getPosition('FORM'))
]
params = {
'window_size': args.window,
'name_model': args.model,
'word_embeddings_file': args.word_embeddings,
'reader_file': args.reader_file,
'num_layers': args.num_layers
}
classifier = WordEmbeddingsClassifier(reader, extractors, ESTIMATORS[args.type], **params)
predicted = classifier.predict()
print >> sys.stderr, len(predicted), len(lines)
labels_idx_rev = {v:k for k,v in reader.vocabulary[reader.getPosition('LABEL')].items()}
i = 0
for line in lines:
line = line.strip()
if line:
print '%s\t%s\t%s' % (line.split()[0], line.split()[1], labels_idx_rev[predicted[i]])
i += 1
else:
print