From d85710499b065a1ee538eefa86003edf01e8da73 Mon Sep 17 00:00:00 2001 From: AytugAltin Date: Thu, 28 Apr 2022 17:49:20 +0200 Subject: [PATCH 1/3] best model --- requirements.txt | 14 +++++++------- train.py | 38 +++++++++++++++++++------------------- 2 files changed, 26 insertions(+), 26 deletions(-) diff --git a/requirements.txt b/requirements.txt index 07ae22e6..4c03d192 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ -joblib==0.16.0 -numpy==1.19.1 -python-dateutil==2.8.1 -pytz==2020.1 -scipy==1.5.2 -six==1.15.0 -threadpoolctl==2.1.0 +joblib==0.16.0 +numpy==1.19.1 +python-dateutil==2.8.1 +pytz==2020.1 +scipy==1.5.2 +six==1.15.0 +threadpoolctl==2.1.0 diff --git a/train.py b/train.py index 2dcb69be..f00ee67e 100644 --- a/train.py +++ b/train.py @@ -1,19 +1,19 @@ -from sklearn.datasets import make_regression -from sklearn.linear_model import LinearRegression -import pickle -import os - -X, y = make_regression(10000,n_features = 10) - -# Train a model -reg = LinearRegression().fit(X, y.ravel()) -# Print out training r2 -print(reg.score(X,y.ravel() )) - -# Write the model to a file -if not os.path.isdir("models/"): - os.mkdir("models") - -filename = 'models/model.pkl' -pickle.dump(reg, open(filename, 'wb')) - +from sklearn.datasets import make_regression +from sklearn.linear_model import Lasso +import pickle +import os + +X, y = make_regression(10000,n_features = 10) + +# Train a model +reg = LinearRegression().fit(X, y.ravel()) +# Print out training r2 +print(reg.score(X,y.ravel() )) + +# Write the model to a file +if not os.path.isdir("models/"): + os.mkdir("models") + +filename = 'models/model.pkl' +pickle.dump(reg, open(filename, 'wb')) + From 17f26a9149ca4c55bee6cafff8cf7291bdb1d46d Mon Sep 17 00:00:00 2001 From: Aytug Date: Thu, 28 Apr 2022 17:52:12 +0200 Subject: [PATCH 2/3] Update requirements.txt --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 4c03d192..6fb92125 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,3 +5,4 @@ pytz==2020.1 scipy==1.5.2 six==1.15.0 threadpoolctl==2.1.0 +scikit-learn From c4041b65dd8557c229c9cd4305276a1aacc6ac4a Mon Sep 17 00:00:00 2001 From: AytugAltin Date: Thu, 28 Apr 2022 17:55:53 +0200 Subject: [PATCH 3/3] added features --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index f00ee67e..793e2cd7 100644 --- a/train.py +++ b/train.py @@ -3,7 +3,7 @@ import pickle import os -X, y = make_regression(10000,n_features = 10) +X, y = make_regression(10000,n_features = 11) # Train a model reg = LinearRegression().fit(X, y.ravel())