| 57 | return "logloss"; |
| 58 | } |
| 59 | void PredTransform(HostDeviceVector<float> *io_preds) const override { |
| 60 | // transform margin value to probability. |
| 61 | std::vector<float> &preds = io_preds->HostVector(); |
| 62 | for (auto& pred : preds) { |
| 63 | pred = 1.0f / (1.0f + std::exp(-pred)); |
| 64 | } |
| 65 | } |
| 66 | void ProbToMargin(linalg::Vector<float>* base_score) const override { |
| 67 | // transform probability to margin value |
| 68 | auto h_intercept = base_score->HostView(); |
no outgoing calls