Open huangXH821 opened 1 year ago
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
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])