regression

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

Bases: aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric

Model prediction MSE

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

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

calculate_metric()[source]

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)[source]

Bases: aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric

Model prediction MAE

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

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

calculate_metric()[source]

Perform metric calculation and return it from this function

Returns

return metric_result

Return type

float or dict