From 3d6d32a5296ae900a653633ba22fb585d80607af Mon Sep 17 00:00:00 2001 From: lukas leufen <l.leufen@fz-juelich.de> Date: Wed, 11 Dec 2019 10:06:08 +0100 Subject: [PATCH] added simple ols model --- requirements.txt | 1 + src/model_modules/linear_model.py | 10 ++++++++++ 2 files changed, 11 insertions(+) create mode 100644 src/model_modules/linear_model.py diff --git a/requirements.txt b/requirements.txt index d2a7200b..ce0e52bb 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,3 +10,4 @@ pytest-cov pytest-html pydot mock +statsmodels \ No newline at end of file diff --git a/src/model_modules/linear_model.py b/src/model_modules/linear_model.py new file mode 100644 index 00000000..9d7ba7e1 --- /dev/null +++ b/src/model_modules/linear_model.py @@ -0,0 +1,10 @@ +__author__ = "Felix Kleinert, Lukas Leufen" +__date__ = '2019-12-11' + + +import statsmodels.api as sm + + +def ordinary_least_squared_model(x, y): + ols_model = sm.OLS(y, x) + return ols_model.fit() -- GitLab