Graph-Learning-Benchmarks / gli

🗂 Graph Learning Indexer: a contributor-friendly and metadata-rich platform for graph learning benchmarks. Dataloading, Benchmarking, Tagging, and more!
https://graph-learning-benchmarks.github.io/gli/
MIT License
42 stars 20 forks source link

refactor save regression and classification tasks #439

Closed jiaqima closed 1 year ago

jiaqima commented 1 year ago

Description

Refactor save_task_node_classification and save_task_node_regression to merge the common code into a private function _save_task_reg_or_cls.

How Has This Been Tested?

The examples in the docstring are tested successfully.

Screenshots (if appropriate):

Screen Shot 2023-04-18 at 10 15 08 PM