diff --git a/CHANGELOG.rst b/CHANGELOG.rst index d039899..96448cc 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -18,6 +18,7 @@ Changelog Ver 0.1.* --------- +* |Feature| |API| Add the general :class:`Metrics` module and `metric_name` parameter for :meth:`evaluate` | `@xuyxu `__ * |Feature| |API| Add :class:`NeuralForestClassifier` and :class:`NeuralForestRegressor` | `@xuyxu `__ * |Fix| Relax check on input dataloader | `@xuyxu `__ * |Feature| |API| Support arbitrary training criteria for all ensembles except Gradient Boosting | `@by256 `__ and `@xuyxu `__ diff --git a/torchensemble/utils/metric.py b/torchensemble/utils/metric.py new file mode 100644 index 0000000..e69de29