Préparer l'entraînement avec Estimators
Dans cet exercice, nous revenons au jeu de données des transactions immobilières du comté de King de la leçon 2. Nous allons de nouveau élaborer et entraîner un modèle de Machine Learning pour prédire les prix des maisons; toutefois, cette fois, nous utiliserons l'API estimator.
Plutôt que de tout faire en une seule étape, nous allons diviser la procédure en plusieurs parties. 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é sous le nom np, et le jeu de données immobilier du comté de King est disponible sous forme de DataFrame pandas : housing.
Cette activité fait partie du cours
Introduction à TensorFlow en Python
Instructions de l’exercice
- Complétez la colonne de caractéristique pour
bedroomset ajoutez une autre colonne de caractéristique 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. - Assignez à
labelsla colonnepricedehousing. - Complétez l'entrée
bedroomsdu dictionnairefeatureset ajoutez une autre entrée pourbathrooms.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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