Skip to content

Commit c4f6fdf

Browse files
authored
fix(GBM): missing functionality of use_reduction_sum (#68)
* Update gradient_boosting.py * Update CHANGELOG.rst
1 parent 7f16b8f commit c4f6fdf

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

CHANGELOG.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ Changelog
44
Ver 0.1.*
55
---------
66

7+
* |Fix| Fix missing functionality of ``use_reduction_sum`` for :meth:`fit` of Gradient Boosting | `@xuyxu <https://github.com/xuyxu>`__
78
* |Enhancement| Relax :mod:`tensorboard` as a soft dependency | `@xuyxu <https://github.com/xuyxu>`__
89
* |Enhancement| |API| Simplify the training workflow of :class:`FastGeometricClassifier` and :class:`FastGeometricRegressor` | `@xuyxu <https://github.com/xuyxu>`__
910
* |Feature| |API| Support TensorBoard logging in :meth:`set_logger` | `@zzzzwj <https://github.com/zzzzwj>`__

torchensemble/gradient_boosting.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -427,6 +427,7 @@ def fit(
427427
super().fit(
428428
train_loader=train_loader,
429429
epochs=epochs,
430+
use_reduction_sum=use_reduction_sum,
430431
log_interval=log_interval,
431432
test_loader=test_loader,
432433
early_stopping_rounds=early_stopping_rounds,
@@ -538,6 +539,7 @@ def fit(
538539
super().fit(
539540
train_loader=train_loader,
540541
epochs=epochs,
542+
use_reduction_sum=use_reduction_sum,
541543
log_interval=log_interval,
542544
test_loader=test_loader,
543545
early_stopping_rounds=early_stopping_rounds,

0 commit comments

Comments
 (0)