After following your tutorial here, I’m trying to figure out how to make a prediction on one node (as if I was presenting an out of sample node and wanted to predict its category).
I feel I’m probably doing something very wrong but I can’t work it out.
This is what I am doing:
print(X.shape) # torch.Size([2708, 1433])
one_node = X[[0]]
print(one_node.shape) # torch.Size([1, 1433])
model(one_node)
Last line errors with:
RuntimeError: SpMM: Invalid input shapes. sparse_mat: [2708, 2708], sparse_val: [99596], dense_mat: [1, 16]. Valid input shapes (sparse_mat, dense_mat) are: (1) (n, m) and (m, k); (2) (n, m) and (m,); (3) (n, m, b) and (m, k, b).
I can’t figure out from that error the correct Tensor shape I’m supposed to pass in. Would really appreciate an example of what it looks like to then predict on the category of an out of sample node.