text_plain_output_1.png image_output_1.png
import matplotlib.pyplot as plt

import pandas as pd

import seaborn as sns

train_data = pd.read_csv('../input/tabular-playground-series-mar-2021/train.csv')
test_data = pd.read_csv('../input/tabular-playground-series-mar-2021/test.csv')

all_data = pd.concat([train_data, test_data])
all_data

train_data.shape

all_data.drop('id', axis=1, inplace=True)
train_data.drop('id', axis=1, inplace=True)
test_data.drop('id', axis=1, inplace=True)

train_data.isnull().sum()

corr_matrix = train_data.corr()

numerics = ['int16', 'int32', 'int64', 'float16', 'float32', 'float64']
num_columns = [col for col in train_data.columns if train_data[col].dtype in numerics and col != 'target']
num_columns