Hitung confusion matrix
Seperti yang Anda pelajari di video, model regresi logistik menghasilkan dua jenis prediksi: prediksi bernilai kontinu berupa probabilitas, dan prediksi kelas yang, dalam contoh himpunan data wells, merupakan kategori diskret dengan dua kelas.
Pada latihan sebelumnya Anda menghitung prediksi bernilai kontinu dalam bentuk probabilitas. Pada latihan ini Anda akan menggunakan nilai tersebut untuk menetapkan kelas pada setiap observasi dalam sampel wells_test Anda. Terakhir, Anda akan mendeskripsikan model menggunakan confusion matrix.
Prediksi yang sudah dihitung prediction dan wells_test sudah dimuat di ruang kerja Anda.
Latihan ini adalah bagian dari kursus
Generalized Linear Models di Python
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Define the cutoff
cutoff = ____
# Compute class predictions: y_prediction
y_prediction = np.where(____ > ____, 1, 0)