how to use this model for molecule property prediction

lucidrains / equiformer-pytorch

Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding

MIT License

242 stars 22 forks source link

how to use this model for molecule property prediction #19

Open chengyunzhang opened 3 months ago

chengyunzhang commented 3 months ago

if possible，please provide a demo。Until now， based on my dataset,i only get the type0 and type1,and i have no idea how to use this model for my molecule property prediction. can you provide a demo?it may really help me !thank you !

zhoufy20 commented 6 days ago

@chengyunzhang Hi, there~Did you solve the problem? I am trying to use it for atomic force predition too.