RegressionLossFunc

class graphstorm.model.RegressionLossFunc

Bases: GSLayer

Loss function for regression tasks.

The torch.nn.MSELoss is used.

forward(logits, labels)

The forward function.

Parameters

logits: torch.Tensor

The prediction results.

labels: torch.Tensor

The training labels.

Returns

loss: Tensor

The loss value.

property in_dims

The number of input dimensions.

Returns

int : the number of input dimensions.

property out_dims

The number of output dimensions.

Returns

int : the number of output dimensions.