mirror of
https://github.com/wassname/Castor.git
synced 2026-09-28 13:42:00 +08:00
Salman Mohammed
d12a9cb475
Initial implementation of RNNs for relation prediction described by Ture and Jojic: https://arxiv.org/abs/1606.05029
23 lines
1.2 KiB
Python
23 lines
1.2 KiB
Python
from argparse import ArgumentParser
|
|
|
|
def get_args():
|
|
parser = ArgumentParser(description='Simple QA BiGRU model - Ferhan')
|
|
parser.add_argument('--epochs', type=int, default=30)
|
|
parser.add_argument('--batch_size', type=int, default=256)
|
|
parser.add_argument('--d_embedding', type=int, default=300)
|
|
parser.add_argument('--d_hidden', type=int, default=300)
|
|
parser.add_argument('--n_layers', type=int, default=1)
|
|
parser.add_argument('--test', action='store_true', dest='test', help='turn on test mode; no training.')
|
|
parser.add_argument('--not_bidirectional', action='store_false', dest='birnn')
|
|
parser.add_argument('--clip', type=float, default=0.25, help='gradient clipping')
|
|
parser.add_argument('--log_every', type=int, default=50)
|
|
parser.add_argument('--lr', type=float, default=1e-3)
|
|
parser.add_argument('--dev_every', type=int, default=300)
|
|
parser.add_argument('--save_every', type=int, default=1000)
|
|
parser.add_argument('--dropout_prob', type=int, default=0.2)
|
|
parser.add_argument('--gpu', type=int, default=0)
|
|
parser.add_argument('--save_path', type=str, default='saved_checkpoints')
|
|
parser.add_argument('--resume_snapshot', type=str, default='')
|
|
args = parser.parse_args()
|
|
return args
|