MulaiMulai sekarang secara gratis

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

Lihat Kursus

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])
Edit dan Jalankan Kode