In [4]:
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
In [9]:
data = pd.read_csv('data.csv')
X = np.array(data[['x1', 'x2']])
y = np.array(data['y'])
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
In [ ]: