diff --git a/slope-evaluator/app/trend_slope_computer.py b/slope-evaluator/app/trend_slope_computer.py index 5b9d5caf9ce9e623df96b2e85aed1ef1bb95ffc0..631a1217d48ea034106b9eb88e8a2ed7f9f579f8 100644 --- a/slope-evaluator/app/trend_slope_computer.py +++ b/slope-evaluator/app/trend_slope_computer.py @@ -9,9 +9,9 @@ def compute(data, warmup_sec): print(regress) - X = regress.iloc[:, 'timestamp'].values.reshape(-1, 1) # values converts it into a numpy array + X = regress.iloc[:, 2].values.reshape(-1, 1) # values converts it into a numpy array print(X) - Y = regress.iloc[:, 'value'].values.reshape(-1, 1) # -1 means that calculate the dimension of rows, but have 1 column + Y = regress.iloc[:, 3].values.reshape(-1, 1) # -1 means that calculate the dimension of rows, but have 1 column print(Y) linear_regressor = LinearRegression() # create object for the class linear_regressor.fit(X, Y) # perform linear regression