Membuat prediksi
Jaringan yang telah dilatih dari latihan kode sebelumnya kini tersimpan sebagai model. Data baru untuk membuat prediksi disimpan dalam array NumPy sebagai pred_data. Gunakan model untuk membuat prediksi pada data baru Anda.
Dalam latihan ini, prediksi Anda berupa probabilitas, yang merupakan cara paling umum bagi data scientist untuk menyampaikan prediksi kepada rekan kerja.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan Python
Petunjuk latihan
- Buat prediksi Anda menggunakan metode
.predict()milik model padapred_data. - Gunakan indeks NumPy untuk menemukan kolom yang sesuai dengan probabilitas prediksi bertahan hidup bernilai True. Ini adalah kolom kedua (indeks
1) daripredictions. Simpan hasilnya dalampredicted_prob_truedan cetak.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Specify, compile, and fit the model
model = Sequential()
model.add(Dense(32, activation='relu', input_shape = (n_cols,)))
model.add(Dense(2, activation='softmax'))
model.compile(optimizer='sgd',
loss='categorical_crossentropy',
metrics=['accuracy'])
model.fit(predictors, target)
# Calculate predictions: predictions
predictions = ____
# Calculate predicted probability of survival: predicted_prob_true
predicted_prob_true = ____
# Print predicted_prob_true
print(predicted_prob_true)