Preparar la formación con Estimadores
Para este ejercicio, volveremos al conjunto de datos de transacciones de vivienda del condado de King del capítulo 2. Volveremos a desarrollar y entrenar un modelo de aprendizaje automático para predecir los precios de la vivienda; sin embargo, esta vez, lo haremos utilizando el estimator
API.
En lugar de completarlo todo en un solo paso, dividiremos este procedimiento en partes. Empezaremos definiendo las columnas de características y cargando los datos. En el siguiente ejercicio, definiremos y entrenaremos un estimator
prefabricado. Ten en cuenta que feature_column
se ha importado para ti desde tensorflow
. Además, numpy
se ha importado como np
, y el conjunto de datos sobre vivienda del condado de Kings está disponible como pandas
DataFrame
: housing
.
Este ejercicio forma parte del curso
Introducción a TensorFlow en Python
Instrucciones de ejercicio
- Completa la columna de características para
bedrooms
y añade otra columna de características numéricas parabathrooms
. Utilizabedrooms
ybathrooms
como claves. - Crea una lista de las columnas de características,
feature_list
, en el orden en que se definieron. - Configura
labels
para que sea igual a la columnaprice
dehousing
. - Completa la entrada
bedrooms
del diccionariofeatures
y añade otra entrada parabathrooms
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Define feature columns for bedrooms and bathrooms
bedrooms = feature_column.numeric_column("____")
bathrooms = ____
# Define the list of feature columns
feature_list = [____, ____]
def input_fn():
# Define the labels
labels = np.array(____)
# Define the features
features = {'bedrooms':np.array(housing['____']),
'bathrooms':____}
return features, labels