Trainer

hivegraph.engine.trainer.AutoTrainer(task: str, **kwargs)[source]

Helper function to instantiate a trainer based on the task.

Parameters:
  • task (str) – Task to be performed by the trainer.

  • **kwargs – Additional keyword arguments needed to instantiate the trainer.

Returns:

Instantiated trainer.

Raises:

ValueError – If the task is not supported.