문제
결정 트리는 데이터를 가능한 한 순수한 그룹으로 나눕니다. 지니 불순도는 그룹의 레이블이 얼마나 섞여 있는지 측정합니다.
Gini = 1 - Σ pₖ²
여기서 pₖ는 클래스 k인 레이블의 비율입니다. 클래스가 하나뿐인 그룹의 불순도는 0이고, 두 클래스가 고르게 섞인 그룹의 불순도는 0.5입니다.
gini_impurity(labels)를 구현하세요.
입력
labels: 그룹 안의 클래스 레이블.
출력
지니 불순도를 실수로 반환하세요.
제약 조건
1 ≤ 길이 ≤ 10,000
각 레이블은 1~20자
예시
예시 1
입력:
labels = ["yes", "yes", "no", "no"]
예상 출력:
expected = 0.5입력:
std::vector<std::string> labels = {"yes", "yes", "no", "no"};
예상 출력:
double expected = 0.5;입력:
string[] labels = new string[] {"yes", "yes", "no", "no"};
예상 출력:
double expected = 0.5;설명
절반은 yes, 절반은 no입니다: 1 - (0.5² + 0.5²) = 1 - 0.5 = 0.5. 두 클래스 그룹이 가장 많이 섞인 상태입니다.
예시 2
입력:
labels = ["cat", "cat", "cat"]
예상 출력:
expected = 0.0입력:
std::vector<std::string> labels = {"cat", "cat", "cat"};
예상 출력:
double expected = 0.0;입력:
string[] labels = new string[] {"cat", "cat", "cat"};
예상 출력:
double expected = 0.0;설명
클래스가 하나뿐: 1 - 1² = 0. 그룹이 순수합니다.
참고 자료
이 문제는 CodeUtility에 맞게 수정되었습니다. 위의 설명과 제한 조건을 따르세요.