Source code for policy_modules.calculate_regression

#!/usr/bin/env python
import numpy as np
from utility_modules.get_parameters import get_params

class FplRegression:  
    def __init__(self, policy_dict):
        fpl_gid= policy_dict['fpl']
        fpl= get_params(fpl_gid)
    def calculate_regression_line(self, x, y, output):
        x_mean= np.mean(x)
        y_mean = np.mean(y)
        x_diff= x-x_mean
        y_diff= y-y_mean
        x_diff_square= np.square(x_diff)
        slope= np.dot(x_diff, y_diff)/np.sum(x_diff_square)
        constant= y_mean - slope*x_mean    
        predicted_y = constant + np.multiply(slope,output)
        return predicted_y