ImadDabbura · August 3, 2018 20:26 · ImadDabbura · Aug 3, 2018
diff --git a/prep_employee_data.py b/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)
No results found