Skip to content

Conversation

@shanghuiyang
Copy link

This patch fixes the transformation while loading an xgboost model file. When the model file is saved from an xgboost model with param "objective= 'multi:softprob'" or "objective= 'multi:softmax'", the loaded model should use TransformSoftmax rather than TransformRaw. This patch works well on my xgboost model.

@coveralls
Copy link

coveralls commented Dec 2, 2020

Pull Request Test Coverage Report for Build 165

  • 1 of 2 (50.0%) changed or added relevant lines in 1 file are covered.
  • No unchanged relevant lines lost coverage.
  • Overall coverage decreased (-0.05%) to 68.6%

Changes Missing Coverage Covered Lines Changed/Added Lines %
xgensemble_io.go 1 2 50.0%
Totals Coverage Status
Change from base Build 159: -0.05%
Covered Lines: 1798
Relevant Lines: 2621

💛 - Coveralls

@nikolaydubina
Copy link

cc: @dmitryikh

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants