search

mdeff / cnn_graph

Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering
https://arxiv.org/abs/1606.09375
MIT License
1.33k stars 390 forks source link

node classification task #22

Open yeiamx opened 6 years ago

yeiamx commented 6 years ago

It seems that it's doing signal classification / regression in the notebook 'usage' . What change should I make to let it do node classification / regression? Only change the matrix A?

mdeff commented 6 years ago

It was indeed developed with signal classification in mind. For node classification, you mainly have to change the objective function. The adjacency matrix A only depends on the graph, whatever the task.