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.KDTree or sklearn.neighbors.BallTree binary 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.