RegressionMixin

class torchgeo.trainers.RegressionMixin(*args, **kwargs)

Bases: LightningModule

Mix-in for regression-based tasks.

Added in version 0.10.

configure_losses()

Initialize the loss criterion.

configure_metrics()

Initialize the performance metrics.

• MeanSquaredError (MSE) and its square root (RMSE). Lower is better.

• MeanAbsoluteError (MAE). Lower is better.

on_train_epoch_end()

Log train metrics.

on_validation_epoch_end()

Log validation metrics.

on_test_epoch_end()

Log test metrics.