text_plain_output_1.png
import numpy as np

import pandas as pd

hr = pd.read_csv(DIR_DATA + '/HR_comma_sep.csv')

hr.sales = hr.sales.astype('category').cat.codes
hr.salary = hr.salary.astype('category').cat.codes
hr[['sales', 'salary']].head()