Yet Another Relation Extraction framework, based on SciIE architecture.

Getting Started

To start to train the network:

python -m yarx train training_config/YOUR_CONFIG.jsonnet -s logs/SAVE_DIR

To continue training:

python -m yarx train logs/SAVE_DIR/config.json -s logs/SAVE_DIR --recover

To evaluate on a dataset:

python -m yarx evaluate logs/SAVE_DIR PATH/TO/DATASET.json

To run inference:

python -m yarx predict logs/SAVE_DIR --silent PATH/TO/DATASET.json --output-file PATH/TO/OUTPUT.jsonlines --batch-size BATCH_SIZE --use-dataset-reader --predictor sciie