UCDP-AEC

Models and data for the UCDP Abstractive Event analysis Corpus paper
git clone https://esimon.eu/repos/UCDP-AEC.git
Log | Files | Refs | README | LICENSE

config_degree_eae_ucdp.json (1071B)


      1 {
      2     "dataset": "ucdp",
      3     "gpu_device": 0,
      4     "seed": 0,
      5     "train_file": "./processed_data/ucdp_bart/train.oneie.json",
      6     "dev_file": "./processed_data/ucdp_bart/dev.oneie.json",
      7     "test_file": "./processed_data/ucdp_bart/test.oneie.json",
      8     "finetune_dir": "./data/degree_eae_ucdp/",
      9     "train_finetune_file": "./data/degree_eae_ucdp/train_all.pkl",
     10     "dev_finetune_file": "./data/degree_eae_ucdp/dev_all.pkl",
     11     "test_finetune_file": "./data/degree_eae_ucdp/test_all.pkl",
     12     "vocab_file": "./data/degree_eae_ucdp/vocab.json",
     13     "output_dir": "./output/degree_eae_ucdp/",
     14     "cache_dir": "./cache",
     15     "model_name": "facebook/bart-large",
     16     "input_style": ["event_type_sent", "template"],
     17     "output_style": ["argument:sentence"],
     18     "max_epoch": 45,
     19     "warmup_epoch": 5,
     20     "train_batch_size": 6,
     21     "eval_batch_size": 12,
     22     "accumulate_step": 3,
     23     "learning_rate": 1e-05,
     24     "weight_decay": 1e-05,
     25     "grad_clipping": 5.0,
     26     "beam_size": 1,
     27     "max_length": 512,
     28     "max_output_length": 150,
     29     "ignore_first_header": false
     30 }