Implement multi-class classification

Created by: asogaard

Currently, the repo support binary graph-level classification. However, for tasks like per-flavour neutrino classification (challenging as it might be), graph-level multi-class would be relevant.