Saber is a deep-learning based tool for information extraction in the biomedical domain. Pull requests are welcome! Note: this is a work in progress. Many things are broken, and the codebase is not stable.
This pull request better integrates the PyTorch based BERT model so that it can be loaded, saved and used for inference (i.e. annotation) under Sabers interface.
The existing Keras based BiLSTM-CRF and the Pytorch based BERT model now have essentially the same interfaces under Saber.
TODO
[x] Write unit tests for as much of the added functionality as possible.
This pull request better integrates the PyTorch based BERT model so that it can be loaded, saved and used for inference (i.e. annotation) under Sabers interface.
The existing Keras based BiLSTM-CRF and the Pytorch based BERT model now have essentially the same interfaces under Saber.
TODO