keras_elmo_bert
keras_elmo_bert copied to clipboard
This repository contains a wrapper for keras models with tf-hub (elmo and bert), which allows you to use them as a keras layer, with the possibility of additional training and masks.
trafficstars
keras_elmo_bert
This repository contains the wrapper of Elmo and Bert models as a keras layer and special tokenizers that allow you to quickly transform raw text into the format you need for these models. Please see example_notebook.ipynb to understand how it works.
The ideas outlined here are a continuation of the remarkable work on the use of tf-hub models from https://github.com/strongio/keras-bert