will-molloy / bert-nlp-java

BERT NLP API for Java with Gradle Tasks to automate training (https://github.com/google-research/bert)

BERT NLP Java

What's included

• Java code to read and process a BERT TensorFlow model
• Gradle tasks to automate the training process

Requirements

• Java 8

Additionally, if you want to train models:

• Python 3.6
• A CUDA GPU (i.e. NVIDIA), with the various software installed
  • Just follow https://www.tensorflow.org/install/gpu
  • Alternatively, access to a Google Cloud TPU

Supported NLP tasks

• Document categorization

WIP:

• Question answering
• Named entity recognition (NER)