fix(GBM): missing functionality of use_reduction_sum (#68)

* Update gradient_boosting.py * Update CHANGELOG.rst
TorchEnsemble-Community · Apr 26, 2021 · c4f6fdf · c4f6fdf
1 parent 7f16b8f
commit c4f6fdf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -4,6 +4,7 @@ Changelog
 Ver 0.1.*
 ---------
 
+* |Fix| Fix missing functionality of ``use_reduction_sum`` for :meth:`fit` of Gradient Boosting | `@xuyxu <https://github.com/xuyxu>`__
 * |Enhancement| Relax :mod:`tensorboard` as a soft dependency | `@xuyxu <https://github.com/xuyxu>`__
 * |Enhancement| |API| Simplify the training workflow of :class:`FastGeometricClassifier` and :class:`FastGeometricRegressor` | `@xuyxu <https://github.com/xuyxu>`__
 * |Feature| |API| Support TensorBoard logging in :meth:`set_logger` | `@zzzzwj <https://github.com/zzzzwj>`__

diff --git a/torchensemble/gradient_boosting.py b/torchensemble/gradient_boosting.py
@@ -427,6 +427,7 @@ def fit(
         super().fit(
             train_loader=train_loader,
             epochs=epochs,
+            use_reduction_sum=use_reduction_sum,
             log_interval=log_interval,
             test_loader=test_loader,
             early_stopping_rounds=early_stopping_rounds,
@@ -538,6 +539,7 @@ def fit(
         super().fit(
             train_loader=train_loader,
             epochs=epochs,
+            use_reduction_sum=use_reduction_sum,
             log_interval=log_interval,
             test_loader=test_loader,
             early_stopping_rounds=early_stopping_rounds,