CoolGraph Versions Save

Make GNN easy to start with

latest

6 months ago

First release of CoolGraph:

  • quick start in one line of code
  • good quality of base models, comparable to state of the art
  • heterogeneous graph support
  • the best model architecture automatic search via Optuna
  • tracking experiments with MLFlow
  • user could define the targets count and target weights in the loss function
  • estimating batch size and neighbourhood sampling sizes for the first and second hop via graph metrics calculation