mccube._kernels.tree
BinaryTreePartitioningKernel
ยค
Bases: AbstractPartitioningKernel
Binary tree based particle partitioning.
Warning
This kernel is based on Sci-Kit learn with JAX JIT compatibility provided
through the use of jax.pure_callback.
Attributes:
-
mode(Literal['KDTree', 'BallTree']) โif to use
sklearn.neighbors.KDTreeorsklearn.neighbors.BallTreebinary tree partitioning modes. -
metric(str | DistanceMetric) โwhat metric to use in the construciton of the tree.
-
metric_kwargs(dict) โkey word arguments taken by the metric.