Can not Run Code MIL-ElECTION-CLASSIFICATION

[huangXH821]

I followed all the steps in ReadME.text. But I got the error for running the Election classification part.

Traceback (most recent call last):
File "d:/mil_code/mil_code/mil_election_classification/run_trials.py", line 12, in
run_deepset(num_neib=0, r=0, alg_name='vanilla')
File "d:\mil_code\mil_code\mil_election_classification\deepset_main.py", line 193, in run_deepset
acc_all = get_performance_test(i)
File "d:\mil_code\mil_code\mil_election_classification\deepset_main.py", line 165, in get_performance_test
value = train(epoch)
File "d:\mil_code\mil_code\mil_election_classification\deepset_main.py", line 130, in train
loss_train = bce(output[idx_train], labels[idx_train])

[huangXH821]

File "d:\mil_code\mil_code\mil_election_classification\deepset_main.py", line 130, in train
loss_train = bce(output[idx_train], labels[idx_train])
IndexError: index 279 is out of bounds for dimension 0 with size 222