(y, left, right, loss)
| 31 | |
| 32 | |
| 33 | def xgb_criterion(y, left, right, loss): |
| 34 | left = loss.gain(left["actual"], left["y_pred"]) |
| 35 | right = loss.gain(right["actual"], right["y_pred"]) |
| 36 | initial = loss.gain(y["actual"], y["y_pred"]) |
| 37 | gain = left + right - initial |
| 38 | return gain |
| 39 | |
| 40 | |
| 41 | def get_split_mask(X, column, value): |