Masalah klasifikasi multiclass
Dalam latihan ini, kita melampaui klasifikasi biner untuk membahas masalah multiclass. Masalah multiclass memiliki target yang dapat mengambil tiga nilai atau lebih. Pada himpunan data kartu kredit, variabel pendidikan dapat mengambil 6 nilai berbeda, masing-masing mewakili tingkat pendidikan yang berbeda. Kita akan menggunakannya sebagai target pada latihan ini dan juga memperluas himpunan fitur dari 3 menjadi 10 kolom.
Seperti pada soal sebelumnya, Anda akan mendefinisikan lapisan masukan, lapisan dense, dan lapisan keluaran. Anda juga akan mencetak prediksi model yang belum dilatih, yaitu probabilitas yang diberikan pada setiap kelas. Tensor fitur telah dimuat dan tersedia sebagai borrower_features. Selain itu, operasi constant(), float32, dan keras.layers.Dense() tersedia.
Latihan ini adalah bagian dari kursus
Pendahuluan TensorFlow di Python
Petunjuk latihan
- Definisikan lapisan input sebagai tensor konstan 32-bit menggunakan
borrower_features. - Atur lapisan dense pertama memiliki 10 node keluaran dan fungsi aktivasi
sigmoid. - Atur lapisan dense kedua memiliki 8 node keluaran dan fungsi aktivasi rectified linear unit.
- Atur lapisan keluaran memiliki 6 node keluaran dan fungsi aktivasi yang sesuai.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Construct input layer from borrower features
inputs = ____
# Define first dense layer
dense1 = keras.layers.Dense(____, activation='____')(inputs)
# Define second dense layer
dense2 = ____
# Define output layer
outputs = ____
# Print first five predictions
print(outputs.numpy()[:5])