search

egyptdj / stagin

STAGIN: Spatio-Temporal Attention Graph Isomorphism Network
https://arxiv.org/abs/2105.13495
GNU General Public License v3.0
85 stars 17 forks source link

cannot convert float NaN to integer #15

Open wby920920 opened 1 year ago

wby920920 commented 1 year ago

When I analyze, the following error occurs. How to solve it?

82eafee8c9d0c75b2f32e67d4167fda
egyptdj commented 1 year ago

Dear @wby920920

I am sorry for my late response. I am not sure which part the error is coming from, but would suspect output of the network_count.values() contain float NaN values. Maybe adding a line for replacing all NaN values of the network_count dict to some integer (e.g. 0) before calling ax.pie would fix the problem. I would welcome a pull request if the problems are fixed!

Thank you.