ID3算法是一种决策树学习算法,用于分类问题。它通过计算信息增益来选择最佳特征作为分裂节点。
以下是使用Python实现ID3算法的示例代码:
``` import numpy as np import pandas as pd from collections import Counter
def entropy(target_col): elements,counts = np.unique(target_col,return_counts = True) entropy = np.sum([(-counts[i]/np.sum(counts))*np.log2(counts[i]/np.sum(counts)) for i in range(len(elements))]) return entropy
def InfoGain(data,split_attribute_name,target_name="class"): total_entropy = entropy(data[target_name]) vals,counts= np.unique(data[split_attribute_name],return_counts=True) Weighted_Entropy = np.sum([(counts[i]/np.sum(counts))*entropy(data.where(data[split_attribute_name]==vals[i]).dropna()[target_name]) for i in range(len(vals))]) Information_Gain = total_entropy - Weighted_Entropy return Information_Gain