training

Documentation

Index

• type Training
  • func New(cfg *config.Config, managerClient managerclient.V2, storage storage.Storage) Training

Types

type Training

type Training interface {
	// Train begins training GNN and MLP model.
	Train(context.Context, string, string) error
}

Training defines the interface to train GNN and MLP model.

func New

func New(cfg *config.Config, managerClient managerclient.V2, storage storage.Storage) Training

New returns a new Training.