search

benedekrozemberczki / SimGNN

A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
GNU General Public License v3.0
759 stars 147 forks source link

Batch processing required. #30

Closed Indradyumna closed 3 years ago

Indradyumna commented 3 years ago

Hi.

Is there a version of the code where we can send batched data into the model? Current version works on one graph pair at a time. This is taking too long for larger training data with each data point being fed one at a time into the model via for loop.

Thanks.

benedekrozemberczki commented 3 years ago

I suppose you can write one if you need one.