DeepNER (tensorflow-gpu)

In this repo is a flask wrapper over neural network architectures for named entity recognition (NER) from the paper "Application of a Hybrid Bi-LSTM-CRF model to the task of Russian Named Entity Recognition" https://arxiv.org/pdf/1709.09686.pdf, which is inspired by LSTM+CRF architecture from https://arxiv.org/pdf/1603.01360.pdf.

Original repository.

Changes

• Added support of Tensorflow-gpu 1.14.0
• Fixed problem with RTX cards (RTX-2070)

Dependencies

• Python 3.6

Library installation

pip install .

Using as a service

Install Flask, and proceed with the following code:

from flask import Flask, jsonify, request
import init

app = Flask(__name__)

@app.route('/ner', methods=['POST'])
def add():
    data = request.get_json()
    tokens, tags = init.predict(data['terms'])
    return jsonify({'tokens': tokens, 'tags': tags})

if __name__ == '__main__':
    app.run(host='localhost', port='5000', debug=False)

NOTE: The latter should be refactored as it might be based on installed ner library.