text_plain_output_1.png
import pandas as pd

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

import seaborn as sns

train_data = pd.read_csv('/kaggle/input/nytfug/train.csv')
test_data = pd.read_csv('/kaggle/input/nytfug/test.csv')

train_data.shape

test_data.shape

train_data.columns

train_data.isnull().sum()

test_data.isnull().sum()

drop_cols = ['id', 'obj_ID', 'run_ID', 'MJD']
train_data.drop(columns=drop_cols, inplace=True)
submission_df = pd.DataFrame({'id': test_data.id})
test_data.drop(columns=drop_cols, inplace=True)

sns.countplot(submission_df.label)