Function metrics

functions.py:64–84

(test_pre, test_real)

def metrics(test_pre, test_real):
    eps = 0.01
    MAPE_test_real = test_real
    MAPE_test_pre = test_pre
    MAPE_test_real[np.where(MAPE_test_real == 0)] = MAPE_test_real[np.where(MAPE_test_real == 0)] + eps
    MAPE_test_pre[np.where(MAPE_test_real == 0)] = MAPE_test_pre[np.where(MAPE_test_real == 0)] + eps
    MAPE = mean_absolute_percentage_error(MAPE_test_real, MAPE_test_pre)
    MAE = mean_absolute_error(test_real, test_pre)
    MSE = mean_squared_error(test_real, test_pre)
    RMSE = np.sqrt(MSE)
    R2 = r2_score(test_real, test_pre)
    RAE = np.sum(abs(test_pre - test_real)) / np.sum(abs(np.mean(test_real) - test_real))
    operator = 100
    print('MAPE: {}%'.format(MAPE*operator))
    print('MAE:{}*10^-2'.format(MAE*operator))
    print('MSE:{}*10^-2'.format(MSE*operator))
    print('RMSE:{}*10^-2'.format(RMSE*operator))
    print('R2:{}%'.format(R2*operator))
    print(('RAE:{}%'.format(RAE*operator)))
    output_list = [MSE, RMSE, MAPE, RAE, MAE, R2]
    return output_list


class CreateDataset(Dataset):