Implement node classification

Created by: asogaard

Currently, the repo support binary graph-level classification. However, for tasks like hit clearning, node-level classification would be relevant.