결정 트리 구현

문제 설명

ID3 알고리즘을 사용하여 결정 트리(Decision Tree)를 구현하세요.

ID3 알고리즘

1. 정보 이득(Information Gain) 계산
2. 최대 정보 이득을 가진 속성 선택
3. 해당 속성으로 데이터 분할
4. 재귀적으로 하위 트리 생성

수식

엔트로피: H(S) = -Σ p_i * log2(p_i)
정보 이득: IG(S, A) = H(S) - Σ (|S_v| / |S|) * H(S_v)

요구사항

• 범주형 속성 처리
• 트리 시각화 (텍스트 형태)
• 새 데이터 예측
• 가지치기(pruning) 옵션

평가 기준

• 알고리즘 정확성
• 예측 정확도
• 코드 가독성