Pisahkan fitur dan variabel target
Sekarang setelah Anda membagi data menjadi pelatihan dan pengujian, saatnya melakukan langkah terakhir sebelum melatih model, yaitu memisahkan fitur dan variabel target ke dalam himpunan data yang berbeda. Anda akan menggunakan daftar nama kolom yang sudah dimuat untuk Anda.
Himpunan data utama dimuat sebagai telcom, dan dibagi menjadi himpunan pelatihan dan pengujian yang masing-masing dimuat sebagai DataFrame pandas ke dalam train dan test. Daftar target dan custid berisi nama variabel target dan ID pelanggan. Anda harus membuat daftar cols dengan nama kolom yang tersisa. Silakan jelajahi himpunan data di konsol.
Latihan ini adalah bagian dari kursus
Machine Learning untuk Pemasaran dengan Python
Petunjuk latihan
- Simpan nama kolom
telcomdalam sebuah daftar dengan mengecualikan nama variabel target dan ID pelanggan. - Ekstrak fitur dan target pelatihan.
- Ekstrak fitur dan target pengujian.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Store column names from `telcom` excluding target variable and customer ID
cols = [col for col in ___.columns if col not in ___ + target]
# Extract training features
train_X = train[___]
# Extract training target
train_Y = train[___]
# Extract testing features
test_X = test[___]
# Extract testing target
test_Y = test[___]