Skip to content
This repository was archived by the owner on Mar 4, 2026. It is now read-only.

my favorite model #6

Open
elleobrien wants to merge 4 commits into
base: master
Choose a base branch
from
Open

my favorite model #6

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
399b419
my favorite model
elleobrien Sep 10, 2020
3ac2f30
Update requirements.txt
elleobrien Sep 10, 2020
f760901
fixed # of inputs
elleobrien Sep 10, 2020
ab9217a
Merge branch 'experiment' of github.com:elleobrien/automatic_testing …
elleobrien Sep 10, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file modified models/model.pkl
View file
Open in desktop
Binary file not shown.
1 change: 1 addition & 0 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,3 +5,4 @@ pytz==2020.1
scipy==1.5.2
six==1.15.0
threadpoolctl==2.1.0
scikit-learn
6 changes: 3 additions & 3 deletions train.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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)
X, y = make_regression(10000,n_features = 11)

# 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() ))

Expand Down