diff --git a/.gitconfig b/.gitconfig new file mode 100644 index 00000000..fd5a1032 --- /dev/null +++ b/.gitconfig @@ -0,0 +1,3 @@ +[credential] + helper = store + password = my_token diff --git a/models/model.pkl b/models/model.pkl index b33a4ffb..5f10b839 100644 Binary files a/models/model.pkl and b/models/model.pkl differ diff --git a/train.py b/train.py index 2dcb69be..d1bbd7bf 100644 --- a/train.py +++ b/train.py @@ -1,12 +1,12 @@ from sklearn.datasets import make_regression -from sklearn.linear_model import LinearRegression +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()) +reg = Lasso().fit(X, y.ravel()) # Print out training r2 print(reg.score(X,y.ravel() ))