search

biomed-AI / GraphEC

Apache License 2.0
24 stars 6 forks source link

conversion from numpy array to tensor #3

Open bndav opened 1 week ago

bndav commented 1 week ago

Hello,

To successfully use the program, I had to add the following line: prottrans_feat = torch.from_numpy(prottrans_feat)

Before the line: pre_computed_node_feat = torch.cat([prottrans_feat, dssp_feat], dim=-1)

In both files: EC_number/data.py Active_sites/data.py