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

1 vote314 views1 answer
Sparse categorical cross entropy python
Python
0 vote302 views1 answer
Python get item from set
Python
0 vote316 views1 answer
Factors of a number with memoization
Python
1 vote336 views1 answer
Drop row with condition dataframe
Python
2 votes275 views1 answer
Sqlalchemy check if db exists
Python
1 vote313 views1 answer
Conda specify multiple channels
Python
0 vote300 views1 answer
Return a new RDD by applying a function to each element of this RDD.
Python
0 vote339 views1 answer
Python import colors
Python
1 vote308 views1 answer
Pandas .nlargest
Python
0 vote376 views1 answer
Circular import error
Python