Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created April 16, 2021 11:24
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/4944bb8d71e0ecf3f5a59dd8bfb56ec1 to your computer and use it in GitHub Desktop.
Save amankharwal/4944bb8d71e0ecf3f5a59dd8bfb56ec1 to your computer and use it in GitHub Desktop.
data = data[["G1", "G2", "G3", "studytime", "failures", "absences"]]
predict = "G3"
x = np.array(data.drop([predict], 1))
y = np.array(data[predict])
from sklearn.model_selection import train_test_split
xtrain, xtest, ytrain, ytest = train_test_split(x, y, test_size=0.2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment