Tech with tim

import pandas as pd
import numpy as np
import sklearn
from sklearn import linear_model
from sklearn.utils import shuffle

data = pd.read_csv("student-mat.csv", sep=";")

data = data[["G1", "G2", "G3", "studytime", "failures", "absences"]]

predict = "G3"

X = np.array(data.drop([predict], 1))
y = np.array(data[predict])

x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(X, y, test_size = 0.1)

Source: techwithtim.net

Similar codes

0 vote145 views1 answer
What does features = data.drop(["Survived", "Sex", "Embarked"], axis=1) do in python
Python
3 votes276 views2 answers
List images in directory python
Python
0 vote127 views1 answer
Python lxml get parent
Python
0 vote134 views1 answer
Starting python project
Python
0 vote136 views1 answer
(for in) printing in python
Python
8 votes159 views2 answers
Python os get dir path
Python
0 vote149 views1 answer
Get length of a tuple in python
Python
0 vote146 views1 answer
List Comprehension generate a list
Python
0 vote143 views1 answer
Python open script in new terminal
Python
76 votes146 views6 answers
Main in python
Python