(y, splits)
| 14 | |
| 15 | |
| 16 | def information_gain(y, splits): |
| 17 | splits_entropy = sum( |
| 18 | [f_entropy(split) * (float(split.shape[0]) / y.shape[0]) for split in splits] |
| 19 | ) |
| 20 | return f_entropy(y) - splits_entropy |
| 21 | |
| 22 | |
| 23 | def mse_criterion(y, splits): |
nothing calls this directly
no test coverage detected