计算基尼指数
(self, y, y1, y2)
| 435 | 分类树，在决策书节点选择计算信息增益/基尼指数，在叶子节点选择多数表决。 |
| 436 | """ |
| 437 | def _calculate_gini_index(self, y, y1, y2): |
| 438 | """ |
| 439 | 计算基尼指数 |
| 440 | """ |
| 441 | p = len(y1) / len(y) |
| 442 | gini = calculate_gini(y) |
| 443 | gini_index = gini - p * \ |
| 444 | calculate_gini(y1) - (1 - p) * \ |
| 445 | calculate_gini(y2) |
| 446 | return gini_index |
| 447 | |
| 448 | |
| 449 | def _calculate_information_gain(self, y, y1, y2): |
nothing calls this directly
no test coverage detected