diff --git a/requirements.txt b/requirements.txt index d2a7200b492fc3dc61f84f4432a97b05051ca184..ce0e52bb72da929f8c97c01ef18ceef6cba86bcc 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 0000000000000000000000000000000000000000..9d7ba7e1a65ce49e10da9e4d9f0131cafc35b3d1 --- /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()