utils/errors.py at master · bhelfrecht/utils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#!/usr/bin/python

import numpy as np

def SUP(predicted, true):
    """
        Supremum absolute error

        ---Arguments---
        predicted: vector of predicted property values
        true: vector of true property values

        ---Returns---
        sup: maximum absolute error
    """
    if np.shape(predicted) != np.shape(true):
        sys.exit("Predicted and true vectors not same shape")
    else:
        abs_err = np.abs(predicted-true)
        sup = np.amax(abs_err, axis=0)
    return sup

def MAE(predicted, true):
    """
        Mean absolute error

        ---Arguments---
        predicted: vector of predicted property values
        true: vector of true property values

        ---Returns---
        mae: mean absolute error
    """
    if np.shape(predicted) != np.shape(true):
        sys.exit("Predicted and true vectors not same shape")
    else:
        abs_err = np.abs(predicted-true)
        mae = np.mean(abs_err, axis=0)
    return mae

def RMSE(predicted, true):
    """
        Root mean square error

        ---Arguments---
        predicted: vector of predicted property values
        true: vector of true property values

        ---Returns---
        rmse: root mean squared error
    """
    if np.shape(predicted) != np.shape(true):
        sys.exit("Predicted and true vectors not same shape")
    else:
        rmse = np.sqrt(np.mean(np.power(predicted-true, 2), axis=0))
    return rmse