New features
- add the
augment()function and document thepredict()function (#190).
Improvements and fixes
- Ancestor matrix is now taken into account, making hierarchical classification now effective (#188).
- Fix steady loss due to frozen optimizer after checkpointing to disk when running on "cuda" and "mps" accelerators.
- multiple R files renamed to improve clarity.