feature_engineer.py

combine = [X_train, X_test]
for dataset in combine:
    dataset['ppl_per_household'] = (dataset['population']/dataset['households'])
    dataset['br_per_household'] = (dataset['total_bedrooms']/dataset['households'])
    dataset['rooms_per_household'] = (dataset['total_rooms']/dataset['households'])