如何获取sklearn中的基尼系数

问题描述 投票:2回答:1

我想在sklearn包中,找到一类路径上每个特征的基尼系数例如虹膜数据中。如鸢尾花花瓣长度基尼:0.4,花瓣基尼:0.4。

python scikit-learn gini
1个回答
0
投票

您可以使用Python + numpy计算基尼系数,如下所示:

from typing import List
from itertools import combinations

import numpy as np

def gini(x: List[float]) -> float:
    x = np.array(x, dtype=np.float32)
    n = len(x)
    diffs = sum(abs(i - j) for i, j in combinations(x, r=2))
    return diffs / (2 * n**2 * x.mean())
© www.soinside.com 2019 - 2024. All rights reserved.