search

networkslab / BagGraph

MIT License
9 stars 2 forks source link

Can not Run Code MIL-ElECTION-CLASSIFICATION #1

Open huangXH821 opened 1 year ago

huangXH821 commented 1 year ago

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 commented 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