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 vote284 views1 answer
Python get desktop directory
Python
0 vote266 views3 answers
Dataframe row print
Python
0 vote285 views1 answer
Reportlab python draw line
Python
1 vote284 views1 answer
Liste in python
Python
0 vote294 views1 answer
Python how to convert a list of floats to a list of strings
Python
0 vote265 views3 answers
Pandas check if column type is list
Python
1 vote294 views1 answer
Example of __init__ self
Python
69 votes274 views6 answers
Python online compiler 3.8
Python
0 vote308 views1 answer
Pandas count empty string values
Python
0 vote283 views1 answer
Tab in python
Python