MulaiMulai sekarang secara gratis

Prediksi klasifikasi

Dalam validasi model, sering kali penting untuk mengetahui lebih dari sekadar klasifikasi akhir. Saat memprediksi siapa yang akan menang dalam sebuah pertandingan, kebanyakan orang juga tertarik pada seberapa besar kemungkinan sebuah tim akan menang.

Probability Prediction Meaning
0 < .50 0 Tim Kalah
.50 + 1 Tim Menang

Dalam latihan ini, Anda akan melihat metode .predict() dan .predict_proba() menggunakan himpunan data tic_tac_toe. Metode pertama akan memberikan prediksi apakah Pemain Satu akan memenangkan permainan, dan metode kedua akan memberikan probabilitas Pemain Satu untuk menang. Gunakan rfc sebagai model klasifikasi random forest.

Latihan ini adalah bagian dari kursus

Validasi Model di Python

Lihat Kursus

Petunjuk latihan

  • Buat dua array prediksi: satu untuk nilai klasifikasi dan satu lagi untuk probabilitas terprediksi.
  • Gunakan metode .value_counts() untuk pandas Series guna mencetak jumlah observasi yang ditetapkan ke setiap kelas.
  • Cetak observasi pertama dari probability_predictions untuk melihat bagaimana probabilitas disusun.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Fit the rfc model. 
rfc.fit(X_train, y_train)

# Create arrays of predictions
classification_predictions = rfc.____(X_test)
probability_predictions = rfc.____(X_test)

# Print out count of binary predictions
print(pd.Series(____).____())

# Print the first value from probability_predictions
print('The first predicted probabilities are: {}'.format(____[____]))
Edit dan Jalankan Kode