From 3f438f536409e734535e55dd3fccaec4fd34fc43 Mon Sep 17 00:00:00 2001 From: xuyxu Date: Sun, 10 Apr 2022 22:01:25 +0800 Subject: [PATCH] update code --- CHANGELOG.rst | 1 + torchensemble/utils/metric.py | 0 2 files changed, 1 insertion(+) create mode 100644 torchensemble/utils/metric.py 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