AutoModel

hivegraph.automodel.AutoModel(task: str, model_config: ConfigDict, **kwargs)

Helper function to instantiate a model based on the task and model name.

Parameters:

• task (str) – Task to be performed by the model.

• model_config (ConfigDict) – Model configuration.

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

Returns:

Instantiated model.

Return type:

torch.nn.Module

Raises:

ValueError – If the task or model name is not supported.