DSE-MSU / DeepRobust

A pytorch adversarial library for attack and defense methods on images and graphs
MIT License
957 stars 191 forks source link

About the sparse matrix #152

Open Kevis9 opened 4 months ago

Kevis9 commented 4 months ago

Hi, Thanks for the great work.

I have two questions.

  1. In the prognn.py file, the EstimatedAdj holds an * ndense matrix , while you pass it to GCN, which require a sparse matrix. Could you please explain this ? I want to know whether to use sparse matrix or dense matrix, because dense matrix contains N*N parameters to train which is troublesome.

  2. If you use a dense matrix as parameters to train, I want to know the limits of number of nodes in a graph. Since 1e4 nodes will need 1e8 paramaters to train. I wonder how you will process large-scale graphs?

Thanks for your patience.