Mempersiapkan pelatihan dengan Estimators
Untuk latihan ini, kita akan kembali ke himpunan data transaksi perumahan King County dari bab 2. Kita akan kembali mengembangkan dan melatih model machine learning untuk memprediksi harga rumah; namun, kali ini kita akan melakukannya menggunakan API estimator.
Alih-alih menyelesaikan semuanya dalam satu langkah, kita akan memecah prosedur ini menjadi beberapa bagian. Kita akan mulai dengan mendefinisikan kolom fitur dan memuat data. Pada latihan berikutnya, kita akan mendefinisikan dan melatih estimator siap pakai. Perhatikan bahwa feature_column telah diimpor untuk Anda dari tensorflow. Selain itu, numpy telah diimpor sebagai np, dan himpunan data perumahan Kings County tersedia sebagai DataFrame pandas: housing.
Latihan ini adalah bagian dari kursus
Pendahuluan TensorFlow di Python
Petunjuk latihan
- Lengkapi kolom fitur untuk
bedroomsdan tambahkan satu kolom fitur numerik lagi untukbathrooms. Gunakanbedroomsdanbathroomssebagai kunci. - Buat daftar kolom fitur,
feature_list, sesuai urutan saat didefinisikan. - Setel
labelsagar sama dengan kolompricepadahousing. - Lengkapi entri
bedroomspada kamusfeaturesdan tambahkan entri lain untukbathrooms.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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