Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- cutMapping = {'Fair': 0 , 'Good': 1, 'Very Good': 2, 'Premium': 3, 'Ideal': 4}
- colorMapping = {'J': 0, 'I': 1, 'H': 2, 'G': 3, 'F': 4, 'E': 5, 'D': 6 }
- clarityMapping = {'I1': 0, 'SI2': 1, 'SI1': 2, 'VS2': 3, 'VS1': 4, 'VVS2': 5, 'VVS1': 6, 'IF' : 7}
- data = data.replace({'cut': cutMapping})
- data = data.replace({'color': colorMapping})
- data = data.replace({'clarity': clarityMapping})
- y = data.price.values
- X = data.drop('price', axis = 1)
- data.head()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement