ImadDabbura/prep_employee_data.py

## prep_employee_data.py
# Rename sales feature into department
df = df.rename(columns={"sales": "department"})
# Map salary into integers
salary_map = {"low": 0, "medium": 1, "high": 2}
df["salary"] = df["salary"].map(salary_map)
# Create dummy variables for department feature
df = pd.get_dummies(df, columns=["department"], drop_first=True)
	# Rename sales feature into department
	df = df.rename(columns={"sales": "department"})
	# Map salary into integers
	salary_map = {"low": 0, "medium": 1, "high": 2}
	df["salary"] = df["salary"].map(salary_map)
	# Create dummy variables for department feature
	df = pd.get_dummies(df, columns=["department"], drop_first=True)