scrub2.py

# Fill null values with each column mean

df = df.fillna(df.mean())

# Define X and y

X = df.drop('y', axis=1)
y = df.y

# Visualize class distribution

y.value_counts().plot(kind='bar')
plt.xticks([0,1], ['No Seizure', 'Seizure'])
plt.ylabel('Count');