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