search

seongjunyun / Graph_Transformer_Networks

Graph Transformer Networks (Authors' PyTorch implementation for the NeurIPS 19 paper)
960 stars 179 forks source link

ImportError: cannot import name 'f1_score' from 'torch_geometric.utils' #41

Open mrwrongwrongwrong opened 1 year ago

mrwrongwrongwrong commented 1 year ago

Is there an update with modified name of f1_score from the torch_geometric?

SaeidKalantari commented 1 year ago

You can just modify the main.py code like this: line 8: from torch_geometric.utils import add_self_loops line 10: from utils import init_seed, _norm, f1_score Remove f1_score from line 8 and import it from utils file in line 10.