awslabs / sagemaker-debugger

Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors
Apache License 2.0
161 stars 83 forks source link

fix xgboost hook through xgboost version 1.7 #648

Closed yl-to closed 1 year ago

yl-to commented 1 year ago

Description of changes:

Tests passed

Style and formatting:

I have run pre-commit install && pre-commit run --all-files to ensure that auto-formatting happens with every commit.

Issue number, if available

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

johnbensnyder commented 1 year ago

This looks good. Can you add some notes on what tests you've run?