Préparer l’entraînement avec les Estimators
Pour cet exercice, nous revenons au jeu de données des transactions immobilières du comté de King, vu au chapitre 2. Nous allons à nouveau développer et entraîner un modèle de Machine Learning pour prédire le prix des maisons ; cependant, cette fois, nous utiliserons l’API estimator.
Plutôt que de tout faire en une seule étape, nous allons découper la procédure. Nous commencerons par définir les colonnes de caractéristiques et charger les données. Dans l’exercice suivant, nous définirons et entraînerons un estimator prédéfini. Notez que feature_column a été importé pour vous depuis tensorflow. De plus, numpy a été importé en tant que np, et le jeu de données immobilier du comté de King est disponible sous forme de DataFrame pandas : housing.
Cet exercice fait partie du cours
Introduction à TensorFlow en Python
Instructions
- Complétez la colonne de caractéristique pour
bedroomset ajoutez une autre colonne de type numérique pourbathrooms. Utilisezbedroomsetbathroomscomme clés. - Créez une liste des colonnes de caractéristiques,
feature_list, dans l’ordre où elles ont été définies. - Affectez à
labelsla colonnepricedehousing. - Complétez l’entrée
bedroomsdu dictionnairefeatureset ajoutez une autre entrée pourbathrooms.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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