Độ vẩn đục Gini

Cây quyết định chia dữ liệu thành các nhóm càng thuần càng tốt. Độ vẩn đục Gini đo mức độ trộn lẫn của các nhãn trong một nhóm:

Gini = 1 - Σ pₖ²

với pₖ là tỉ lệ nhãn thuộc lớp k. Nhóm chỉ có một lớp có độ vẩn đục 0; nhóm trộn đều hai lớp có độ vẩn đục 0.5.

Hãy cài đặt gini_impurity(labels).

Đầu vào

labels: Các nhãn lớp trong nhóm.

Đầu ra

Trả về độ vẩn đục Gini dưới dạng số thực.

Ràng buộc

1 ≤ độ dài ≤ 10.000

Mỗi nhãn dài 1 đến 20 ký tự

Ví dụ

Đầu vào:
labels = ["yes", "yes", "no", "no"]

Kết quả mong đợi:
expected = 0.5

Đầu vào:
std::vector<std::string> labels = {"yes", "yes", "no", "no"};

Kết quả mong đợi:
double expected = 0.5;

Đầu vào:
string[] labels = new string[] {"yes", "yes", "no", "no"};

Kết quả mong đợi:
double expected = 0.5;

Đầu vào:
labels = ["cat", "cat", "cat"]

Kết quả mong đợi:
expected = 0.0

Đầu vào:
std::vector<std::string> labels = {"cat", "cat", "cat"};

Kết quả mong đợi:
double expected = 0.0;

Đầu vào:
string[] labels = new string[] {"cat", "cat", "cat"};

Kết quả mong đợi:
double expected = 0.0;

Tham khảo

Thử thách này đã được điều chỉnh cho CodeUtility. Hãy làm theo đề bài và các ràng buộc ở trên.

• scikit-learn, Decision trees: mathematical formulation