add proxies for dataset functions in task api

BorgwardtLab / proteinshake

Protein structure datasets for machine learning.

https://proteinshake.ai

BSD 3-Clause "New" or "Revised" License

99 stars 8 forks source link

Closed timkucera closed 1 year ago

timkucera commented 1 year ago

Added some functions to be able to do this:

from proteinshake.tasks import EnzymeCommissionTask
task = EnzymeCommissionTask(root='./data').to_graph(eps=8).pyg()
print(task.test)

Let me know what you think @cgoliver

cgoliver commented 1 year ago

looks good to me