1. 17 Oct, 2025 1 commit
  2. 14 Oct, 2025 1 commit
  3. 13 Oct, 2025 2 commits
  4. 11 Oct, 2025 1 commit
  5. 23 Sep, 2025 1 commit
  6. 22 Sep, 2025 1 commit
    • Jeff Daily's avatar
      [ROCm] re-add support for ROCm builds · 61ec4f1a
      Jeff Daily authored
      Previously #6086 added ROCm support but after numerous rebases it lost
      critical changes. This PR restores the ROCm build.
      
      There are many source file changes but most were automated using the
      following:
      
      ```bash
      for f in `grep -rl '#ifdef USE_CUDA'`
      do
          sed -i 's@#ifdef USE_CUDA@#if defined(USE_CUDA) || defined(USE_ROCM)@g' $f
      done
      
      for f in `grep -rl '#endif  // USE_CUDA'`
      do
          sed -i 's@#endif  // USE_CUDA@#endif  // USE_CUDA || USE_ROCM@g' $f
      done
      ```
      61ec4f1a
  7. 24 Aug, 2025 1 commit
  8. 24 Jul, 2025 1 commit
  9. 10 Oct, 2023 1 commit
  10. 08 Oct, 2023 1 commit
    • shiyu1994's avatar
      [CUDA] CUDA Quantized Training (fixes #5606) (#5933) · f901f471
      shiyu1994 authored
      * add quantized training (first stage)
      
      * add histogram construction functions for integer gradients
      
      * add stochastic rounding
      
      * update docs
      
      * fix compilation errors by adding template instantiations
      
      * update files for compilation
      
      * fix compilation of gpu version
      
      * initialize gradient discretizer before share states
      
      * add a test case for quantized training
      
      * add quantized training for data distributed training
      
      * Delete origin.pred
      
      * Delete ifelse.pred
      
      * Delete LightGBM_model.txt
      
      * remove useless changes
      
      * fix lint error
      
      * remove debug loggings
      
      * fix mismatch of vector and allocator types
      
      * remove changes in main.cpp
      
      * fix bugs with uninitialized gradient discretizer
      
      * initialize ordered gradients in gradient discretizer
      
      * disable quantized training with gpu and cuda
      
      fix msvc compilation errors and warnings
      
      * fix bug in data parallel tree learner
      
      * make quantized training test deterministic
      
      * make quantized training in test case more accurate
      
      * refactor test_quantized_training
      
      * fix leaf splits initialization with quantized training
      
      * check distributed quantized training result
      
      * add cuda gradient discretizer
      
      * add quantized training for CUDA version in tree learner
      
      * remove cuda computability 6.1 and 6.2
      
      * fix parts of gpu quantized training errors and warnings
      
      * fix build-python.sh to install locally built version
      
      * fix memory access bugs
      
      * fix lint errors
      
      * mark cuda quantized training on cuda with categorical features as unsupported
      
      * rename cuda_utils.h to cuda_utils.hu
      
      * enable quantized training with cuda
      
      * fix cuda quantized training with sparse row data
      
      * allow using global memory buffer in histogram construction with cuda quantized training
      
      * recover build-python.sh
      
      enlarge allowed package size to 100M
      f901f471
  11. 16 Jun, 2023 1 commit
  12. 16 Mar, 2023 1 commit
  13. 01 Feb, 2023 1 commit
    • James Lamb's avatar
      [CUDA] consolidate CUDA versions (#5677) · 4f47547c
      James Lamb authored
      
      
      * [ci] speed up if-else, swig, and lint conda setup
      
      * add 'source activate'
      
      * python constraint
      
      * start removing cuda v1
      
      * comment out CI
      
      * remove more references
      
      * revert some unnecessaary changes
      
      * revert a few more mistakes
      
      * revert another change that ignored params
      
      * sigh
      
      * remove CUDATreeLearner
      
      * fix tests, docs
      
      * fix quoting in setup.py
      
      * restore all CI
      
      * Apply suggestions from code review
      Co-authored-by: default avatarshiyu1994 <shiyu_k1994@qq.com>
      
      * Apply suggestions from code review
      
      * completely remove cuda_exp, update docs
      
      ---------
      Co-authored-by: default avatarshiyu1994 <shiyu_k1994@qq.com>
      4f47547c
  14. 29 Dec, 2022 1 commit
  15. 27 Dec, 2022 1 commit
    • shiyu1994's avatar
      [CUDA] Add L2 metric for new CUDA version (#5633) · 6482b47e
      shiyu1994 authored
      * add rmse metric for new cuda version
      
      * add Init for CUDAMetricInterface
      
      * fix lint errors
      
      * fix rmse and add l2 metric for new cuda version
      
      * use CUDAL2Metric
      
      * explicit template instantiation
      
      * write result only with the first thread
      
      * pre allocate buffer for output converting
      
      * fix l2 regression with cuda metric evaluation
      
      * weighting loss in cuda metric evaluation
      
      * mark CUDATree::AsConstantTree as override
      6482b47e
  16. 02 Dec, 2022 1 commit
  17. 29 Jul, 2022 1 commit
  18. 08 Oct, 2021 1 commit
  19. 05 Oct, 2021 1 commit
  20. 04 May, 2021 1 commit
  21. 17 Mar, 2021 1 commit
    • ashok-ponnuswami-msft's avatar
      Range check for DCG position discount lookup (#4069) · 4580393f
      ashok-ponnuswami-msft authored
      * Add check to prevent out of index lookup in the position discount table. Add debug logging to report number of queries found in the data.
      
      * Change debug logging location so that we can print the data file name as well.
      
      * Revert "Change debug logging location so that we can print the data file name as well."
      
      This reverts commit 3981b34bd6e0530f89c4733e78e6b6603bf50d48.
      
      * Add data file name to debug logging.
      
      * Move log line to a place where it is output even when query IDs are read from a separate file.
      
      * Also add the out-of-range check to rank metrics.
      
      * Perform check after number of queries is initialized.
      
      * Update
      4580393f
  22. 19 Feb, 2021 1 commit
  23. 29 Sep, 2020 1 commit
    • Guolin Ke's avatar
      fix warnings (#3399) · 3c0e12dc
      Guolin Ke authored
      * fix warnings
      
      * Apply suggestions from code review
      
      * Update feature_group.h
      
      * Update feature_group.h
      
      * Update src/treelearner/serial_tree_learner.cpp
      
      * Update multiclass_metric.hpp
      3c0e12dc
  24. 23 Sep, 2020 1 commit
    • Belinda Trotta's avatar
      Average precision score (#3347) · 28704900
      Belinda Trotta authored
      * Implement average precision score
      
      * Fix lint errors
      
      * Change name to average_precision
      
      * Add to R-package list of metrics
      
      * Empty commit to trigger CI jobs
      
      * Change name to average_precision
      28704900
  25. 20 Sep, 2020 1 commit
    • Belinda Trotta's avatar
      Auc mu weights (#3349) · 1782fcb1
      Belinda Trotta authored
      * Update auc_mu metric to use data weights if provided
      
      * Calculate class sizes and total weights in Init so we only do it once
      
      * Fix lint error
      
      * Empty commit to trigger CI jobs
      1782fcb1
  26. 07 Jul, 2020 1 commit
  27. 05 Jun, 2020 1 commit
  28. 01 Jun, 2020 1 commit
  29. 02 Mar, 2020 2 commits
  30. 22 Feb, 2020 1 commit
    • Guolin Ke's avatar
      some code refactoring (#2769) · 3e80df7e
      Guolin Ke authored
      * some refines
      
      * more omp refactoring
      
      * format define
      
      * fix merge bug
      
      * some fixes
      
      * fix some warnings
      
      * Apply suggestions from code review
      
      * Apply suggestions from code review
      
      * remove dup codes
      3e80df7e
  31. 17 Dec, 2019 1 commit
  32. 12 Dec, 2019 1 commit
    • Belinda Trotta's avatar
      auc-mu metric (#2567) · 222775ca
      Belinda Trotta authored
      * Fix bug where small values of max_bin cause crash.
      
      * Revert "Fix bug where small values of max_bin cause crash."
      
      This reverts commit fe5c8e2547057c1fa5750bcddd359dd7708fab4b.
      
      * Add auc-mu multiclass metric.
      
      * Fix bug where scores are equal.
      
      * Merge.
      
      * Change name to auc_mu everywhere (instead of auc-mu).
      
      * Fix comparison between signed and unsigned int.
      
      * Change name to AUC-mu in docs and output messages.
      
      * Improve test.
      
      * Use prefix increment.
      
      * Update R package.
      
      * Fix style issues.
      
      * Tidy up test code.
      
      * Read all lines first then process.
      
      * Allow passing AUC-mu weights directly as a list in parameters.
      
      * Remove unused code, improve example and docs.
      222775ca
  33. 29 Nov, 2019 1 commit
  34. 01 Nov, 2019 1 commit
  35. 22 Sep, 2019 1 commit
  36. 25 Jul, 2019 2 commits
  37. 26 May, 2019 1 commit
    • Belinda Trotta's avatar
      Top k multi error (#2178) · b3db9e92
      Belinda Trotta authored
      * Implement top-k multiclass error metric. Add new parameter top_k_threshold.
      
      * Add test for multiclass metrics
      
      * Make test less sensitive to avoid floating-point issues.
      
      * Change tabs to spaces.
      
      * Fix problem with test in Python 2. Refactor to use np.testing. Decrease number of training rounds so loss is larger and easier to compare.
      
      * Move multiclass tests into test_engine.py
      
      * Change parameter name from top_k_threshold to multi_error_top_k.
      
      * Fix top-k error metric to handle case where scores are equal. Update tests and docs.
      
      * Change name of top-k metric to multi_error@k.
      
      * Change tabs to spaces.
      
      * Fix formatting.
      
      * Fix minor issues in docs.
      b3db9e92