text_plain_output_1.png
from sklearn.model_selection import train_test_split

import gc

import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

data = pd.read_csv('../input/train.csv')
y = data.deal_probability.copy()
X_train, X_test, y_train, y_test = train_test_split(data, y, test_size=0.2, random_state=23)
del data
gc.collect()
X_train.head()