Skip to content

Commit

Permalink
[ENH] Implement Proximity Tree classifier using aeon distances (#1566)
Browse files Browse the repository at this point in the history
* Yet to be wrapped by numba

* Added gini scores

* Initial logic for tree

* get best splitter

* Minor typo

* build_tree method implemented

* predict method

* Completed the proximity tree

* random_state and staticmethod

* predict and predict_proba

* tests for gini and gini_gain

* test parameter value

* Fixing random state

* Fix recursion base case

* Completed API

* Completed API

* Test case pass

* Improve API

* Minor corrections

* Test for accuracy and pytest fixture

* Improve API and code cov

* Comment at the top of the file

* Capability to handle 3d arrays

* Capability to handle 3d arrays

* Added API Reference

---------

Co-authored-by: Matthew Middlehurst <[email protected]>
  • Loading branch information
itsdivya1309 and MatthewMiddlehurst authored Jun 27, 2024
1 parent 72ef471 commit fd27e7f
Show file tree
Hide file tree
Showing 4 changed files with 679 additions and 4 deletions.
6 changes: 2 additions & 4 deletions aeon/classification/distance_based/__init__.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,9 @@
"""Distance based time series classifiers."""

__all__ = [
"ElasticEnsemble",
"KNeighborsTimeSeriesClassifier",
]
__all__ = ["ElasticEnsemble", "KNeighborsTimeSeriesClassifier", "ProximityTree"]

from aeon.classification.distance_based._elastic_ensemble import ElasticEnsemble
from aeon.classification.distance_based._proximity_tree import ProximityTree
from aeon.classification.distance_based._time_series_neighbors import (
KNeighborsTimeSeriesClassifier,
)
Loading

0 comments on commit fd27e7f

Please sign in to comment.