text_plain_output_1.png
from sklearn.model_selection import train_test_split

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

import seaborn as sns

given_data = pd.read_csv('/kaggle/input/ml-olympiad-good-health-and-well-being/train.csv')
test_data = pd.read_csv('/kaggle/input/ml-olympiad-good-health-and-well-being/test.csv')

given_data.columns

given_data.isnull().sum()

test_data.isnull().sum()

drop_cols = ['BMI', 'PatientID']
given_data.drop(columns=drop_cols, inplace=True)
test = test_data.drop(columns=drop_cols)

y = given_data['target']
X = given_data.drop(columns=['target'])

X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)
print(len(X_train), len(X_test))