regression

class aitoolbox.experiment.core_metrics.regression.MeanSquaredErrorMetric(y_true, y_predicted)

Bases: AbstractBaseMetric

Model prediction MSE

Parameters:

• y_true (numpy.ndarray or list) – ground truth targets

• y_predicted (numpy.ndarray or list) – predicted targets

calculate_metric()

Perform metric calculation and return it from this function

Returns:

return metric_result

Return type:

float or dict

class aitoolbox.experiment.core_metrics.regression.MeanAbsoluteErrorMetric(y_true, y_predicted)

Bases: AbstractBaseMetric

Model prediction MAE

Parameters:

• y_true (numpy.ndarray or list) – ground truth targets

• y_predicted (numpy.ndarray or list) – predicted targets

calculate_metric()

Perform metric calculation and return it from this function

Returns:

return metric_result

Return type:

float or dict