from src.data import df, df_norm
from sklearn.naive_bayes import GaussianNB

cls = GaussianNB()
model = cls.fit(df_norm, df['genre'])