Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def get_normalization_parameters(traindf, features):
- """Get the normalization parameters (E.g., mean, std) for traindf for
- features. We will use these parameters for training, eval, and serving."""
- def _z_score_params(column):
- mean = traindf[column].mean()
- std = traindf[column].std()
- return {'mean': mean, 'std': std}
- normalization_parameters = {}
- for column in features:
- normalization_parameters[column] = _z_score_params(column)
- return normalization_parameters
- NUMERIC_FEATURES = ['housing_median_age', 'total_rooms', 'total_bedrooms',
- 'population', 'households', 'median_income']
- normalization_parameters = get_normalization_parameters(traindf, NUMERIC_FEATURES)
Add Comment
Please, Sign In to add comment