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

2 votes292 views1 answer
Show rows with a null value pandas
Python
-1 vote273 views1 answer
Python plt show legend
Python
5 votes315 views1 answer
Date time module date diference
Python
37 votes284 views6 answers
If list element contains string python
Python
1 vote265 views1 answer
Python find smallest value in 2d list
Python
0 vote281 views6 answers
List comprehension
Python
13 votes337 views5 answers
Convert column to numeric pandas
Python
0 vote288 views1 answer
For loop to create a set of counters in python
Python
4 votes283 views2 answers
Median of a list python
Python
0 vote306 views1 answer
Syntax of calloc
Python