tabensemb.data.datamodule.DataModule._infer_loss#

method

DataModule._infer_loss(task: str)[source]#

Automatically infer the loss type using the name of the task and the loss type given in the configuration.

Parameters:
task

“binary”, “multiclass”, or “regression”

Returns:
str

“mse” or “mae” for regression tasks and “cross_entropy” for classification tasks.