Menghitung indeks Gini
Algoritma decision tree bertujuan memperoleh partisi pada node terminal yang semurni mungkin. Indeks Gini adalah salah satu metode yang digunakan untuk mencapainya. Nilai ini dihitung berdasarkan proporsi sampel pada tiap kelompok.
Diberikan jumlah orang yang stayed dan left, hitunglah indeks Gini untuk node tersebut.
Latihan ini merupakan bagian dari kursus
Analitik SDM: Memprediksi Perputaran Karyawan dengan Python
Instruksi latihan
- Hitung total jumlah karyawan pada node tersebut.
- Hitung indeks Gini berdasarkan proporsi karyawan di tiap kelompok.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)