search

bknyaz / graph_attention_pool

Attention over nodes in Graph Neural Networks using PyTorch (NeurIPS 2019)
https://arxiv.org/abs/1905.02850
Other
283 stars 48 forks source link

Threshold-based pooling #1

Closed dchang56 closed 5 years ago

dchang56 commented 5 years ago

Hello!

Is the threshold-based pooling method in your paper pretty much the same as using the "min_score" argument in pytorch geometric's TopKPooling?

bknyaz commented 5 years ago

Hi, exactly.