text_plain_output_1.png image_output_1.png
from sklearn.model_selection import train_test_split

import pandas as pd

train_df = pd.read_csv('../input/tabular-playground-series-sep-2021/train.csv')
test_df = pd.read_csv('../input/tabular-playground-series-sep-2021/test.csv')
sample_df = pd.read_csv('../input/tabular-playground-series-sep-2021/sample_solution.csv')

X, y = (train_df.iloc[:, 1:-1], train_df.iloc[:, -1])
print(X.shape, y.shape)
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=33)
print(X_train.shape, X_test.shape, y_train.shape, y_test.shape)