topobench.evaluator.metrics package#

Init file for custom metrics in evaluator module.

class ExampleRegressionMetric(squared=True, num_outputs=1, **kwargs)#

Example metric.

Parameters:

__init__(squared=True, num_outputs=1, **kwargs)#

Initialize internal Module state, shared by both nn.Module and ScriptModule.

compute()#

Compute mean squared error over state.

Returns:

update(preds, target)#

Update state with predictions and targets.

Parameters:

Submodules#