text_html_output_1.png
from sklearn.model_selection import train_test_split

import pandas as pd

aug_data = pd.read_csv('../input/hr-analytics-job-change-of-data-scientists/aug_train.csv', index_col='enrollee_id')
aug_data = aug_data.sort_index()
aug_data

aug_data.isnull().sum()

aug_data.isnull().sum()

y = aug_data.target.astype('int')
X = aug_data.drop('target', axis=1)
X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.1)
print('X_train: ', X_train.shape, 'y_train: ', y_train.shape, '\nX_val: ', X_val.shape, 'y_val: ', y_val.shape)