Home

Awesome

This is "building gap resolution overnight" repository for Meanotek submission for AGRR competition

Installation

  1. First, install NeuThink library.
  2. Then clone this repository.
  3. Download pre-trained context embedding model from here and put in into the model3000x2x20482 folder

Usage

To Train model:

  1. Convert train data to conll using agrr2conll.py. You will need to change input and output file names in the script
  2. Run train.py (ensure that you have converted file in the same directory). The script requires you to have GPU (8Gb GPU RAM is needed).
  3. Wait for completion
  4. Run predict_gap.py for test file
  5. convert data back to agrr format with conll2agrr.py

please contact dtarasov@meanotek.io if you have any problems