mix nasty.train.coref (Nasty v0.3.0)

Train neural coreference resolution models.

Usage

mix nasty.train.coref \
  --corpus data/ontonotes/train \
  --dev data/ontonotes/dev \
  --output priv/models/en/coref \
  --epochs 20 \
  --batch-size 32 \
  --learning-rate 0.001

Options

• --corpus - Path to training data directory (required)
• --dev - Path to development data directory (required)
• --output - Base path for saving models (required)
• --epochs - Number of training epochs (default: 20)
• --batch-size - Training batch size (default: 32)
• --learning-rate - Learning rate (default: 0.001)
• --hidden-dim - LSTM hidden dimension (default: 128)
• --dropout - Dropout rate (default: 0.3)
• --patience - Early stopping patience (default: 3)
• --max-distance - Max sentence distance for pairs (default: 3)