GSgnnNodeModelBase

class graphstorm.model.GSgnnNodeModelBase(*args, **kwargs)

Bases: GSgnnNodeModelInterface, GSgnnModelBase

GraphStorm base model class for node prediction tasks.

This base class extends GraphStorm GSgnnModelBase and GSgnnNodeModelInterface. When users want to define a customized node prediction GNN model and train the model in GraphStorm, the model class needs to inherit from this base class, and implement the required methods including forward(), predict(), save_model(), restore_model() and create_optimizer().