Support binary:logistic objective and probability predictions#4
Merged
Conversation
扩展模型加载和预测以支持 binary:logistic 目标函数。 - 在 predict_ensemble 中根据任务类型应用 sigmoid 变换 - 在模型加载时解析 objective 字段并设置 PredictionTask - 对 binary:logistic 的 base_score 进行 logit 变换以获取 base_margin - 更新文档和错误信息以反映对回归和分类的支持
- 新增测试以验证从JSON加载的二元分类器能正确预测概率 - 新增测试以验证分类器模型能正确报告基础概率分数 - 移除不再适用的“不支持的目标函数”测试，因为分类器模型现在已被支持
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
增加Binary-二分类模型支持