import pandas as pd
from sklearn.preprocessing import OrdinalEncoder
from sklearn.naive_bayes import CategoricalNB

# "train.csv" is downloaded from the kaggle competition page
# https://www.kaggle.com/c/amazon-employee-access-challenge

amazon = pd.read_csv('train.csv')

y = amazon['ACTION']
x = amazon.drop(columns=['ACTION'])

enc = OrdinalEncoder()
x = enc.fit_transform(x)

clf = CategoricalNB()
clf.fit(x, y)