Running example code for ESL

[tanfiona]

Hi, thanks for sharing your code!

I am trying to use it on ESL. When I run the code chunk in your README, but for ESL:

python src/main.py --const_eval=0 --data_dir=data --data_type=esl --downsample=0.015 --epochs=150 --eval_step=1 --eval_type=two --intersection_temp=0.0005 --lambda_anno=0 --lambda_condi_h=0.1 --lambda_condi_m=1 --lambda_cross=0 --lambda_pair_h=0.4 --lambda_pair_m=0.8 --lambda_trans=0 --learning_rate=0.0001 --log_batch_size=4 --loss_type=4 --lstm_hidden_size=256 --lstm_input_size=768 --max_grad_norm=10 --mlp_size=512 --model=box --model_save=1 --num_layers=1 --patience=20 --proj_output_dim=640 --save_plot=0 --threshold1=-0.3 --threshold2=-0.7 --volume_temp=5

I am faced with an error from data_loader as follows:

Traceback (most recent call last):
  File "/home/fiona/CE2ERE/src/main.py", line 375, in <module>
    main()
  File "/home/fiona/CE2ERE/src/main.py", line 370, in main
    trainer, evaluator = setup(args)
  File "/home/fiona/CE2ERE/src/main.py", line 222, in setup
    train_dataloader, valid_dataloader_dict, test_dataloader_dict, valid_cv_dataloader_dict, test_cv_dataloader_dict, num_classes, n_tags = create_dataloader(args)
  File "/home/fiona/CE2ERE/src/main.py", line 74, in create_dataloader
    esl_train_set, esl_valid_set, esl_test_set, esl_valid_cv_set, esl_test_cv_set = esl_data_loader(args, data_dir)
  File "/home/fiona/CE2ERE/src/data_loader.py", line 456, in esl_data_loader
    train_set = get_hieve_train_set(data_dict, args.downsample, args.model, args.symm_train)
  File "/home/fiona/CE2ERE/src/data_loader.py", line 39, in get_hieve_train_set
    append_hieve_train_dataset(train_set, downsample, model_type, x, y, z, event_dict, sntc_dict, relation_dict)
  File "/home/fiona/CE2ERE/src/data_loader.py", line 64, in append_hieve_train_dataset
    x_sntc_id = event_dict[x]["sent_id"]
KeyError: 1

Any thoughts as to how to fix this? Thank you.