Skip to content
Snippets Groups Projects

Resolve "release v1.4.0"

Merged Ghost User requested to merge release_v1.4.0 into master
1 file
+ 16
2
Compare changes
  • Side-by-side
  • Inline
@@ -196,9 +196,23 @@ def log_apply(data: Data, mean: Data, std: Data) -> Data:
return standardise_apply(np.log1p(data), mean, std)
def mean_squared_error(a, b):
def mean_squared_error(a, b, dim=None):
"""Calculate mean squared error."""
return np.square(a - b).mean()
return np.square(a - b).mean(dim)
def mean_absolute_error(a, b, dim=None):
"""Calculate mean absolute error."""
return np.abs(a - b).mean(dim)
def calculate_error_metrics(a, b, dim):
"""Calculate MSE, RMSE, and MAE. Additionally return number of used values for calculation."""
mse = mean_squared_error(a, b, dim)
rmse = np.sqrt(mse)
mae = mean_absolute_error(a, b, dim)
n = (a - b).notnull().sum(dim)
return {"mse": mse, "rmse": rmse, "mae": mae, "n": n}
class SkillScores:
Loading