laituan245 / AIDA-Entity-Coref

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AIDA Entity Coreference

docker build --tag laituan245/spanbert_entity_coref .
docker push laituan245/spanbert_entity_coref
docker run --gpus '"device=0"' --rm -v /shared:/shared laituan245/spanbert_entity_coref -edl_official input1.tab -edl_freebase input2.tab -l ltf_dir -o output.tab

Training Instructions

To train a model, simply run the following command

python runner.py 

By default, the code will use the spanbert_large config.

About


Languages

Language:Python 98.1%Language:Dockerfile 1.7%Language:Shell 0.2%