torch_geometric.nn.functional.gini
- gini(w: Tensor) Tensor[source]
来自“通过正交化和诱导稀疏性提高分子图神经网络可解释性”论文的基尼系数。
根据以下公式计算矩阵每行的正则化惩罚 \(\in [0, 1]\)
\[\mathcal{L}_\textrm{Gini}^i = \sum_j^n \sum_{j'}^n \frac{|w_{ij} - w_{ij'}|}{2 (n^2 - n)\bar{w_i}}\]并返回所有行的平均值。
- Parameters:
w (torch.Tensor) – 一个二维张量。
- Return type: