Open fatemehghanadi opened 8 months ago
In ndarray, you need to change it to int64. You can modify the data type yourself. I hope it will be helpful to you.
One simple way: train_set, test_set = dataset[train_index.tolist()], dataset[test_index.tolist()]
Hello. when i run this code on cuda environment this error happend. would you please help me fixing it?
python -m examples.example_main --dataset_name ABIDE --pooling concat --gcn_mp_type edge_node_concate --hidden_dim 256