Gini指数是衡量数据的不均衡程度的一种指标,可以用来评估分类模型的效果。在Python中,可以使用numpy和scipy等科学计算库来编写计算Gini指数的代码。以下是一个简单的示例:
import numpy as np
def gini(y_true, y_pred):
# 先统计真实标签和预测标签中不同值的数量
classes = np.unique(y_true)
n_classes = len(classes)
# 初始化Gini指数
gini = 0.0
# 对于每个类别,统计其在真实标签和预测标签中的数量