Skip to content

Instantly share code, notes, and snippets.

@sam-thecoder

sam-thecoder/SGD2.py

Last active August 9, 2018 11:10
Show Gist options
  • Save sam-thecoder/42d1a600811945c2c2bd5662d00cf3d2 to your computer and use it in GitHub Desktop.
Save sam-thecoder/42d1a600811945c2c2bd5662d00cf3d2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
SGD2.py
from sklearn.model_selection import train_test_split
from sklearn import linear_model
y = df['Predict'].values
X = df[['Value USD', 'Drop 7', 'Up 7', 'Mean Change 7', 'Change']].values
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20, random_state=42)
clf = linear_model.SGDClassifier()
clf.fit(X_train, y_train)
clf.score(X_test, y_test)
#0.4381443298969072
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment