xoak.index.sklearn_adapters.SklearnBallTreeAdapter

class xoak.index.sklearn_adapters.SklearnBallTreeAdapter(**kwargs)[source]

Xoak index adapter for sklearn.neighbors.BallTree.

This index adapter is registered in xoak under the name sklearn_balltree. You can use it in xarray.Dataset.xoak.set_index() by simply providing its name for the index_type argument. Alternatively, you can access it via the index registry, i.e.,

>>> import xoak
>>> ireg = xoak.IndexRegistry()
>>> ireg.sklearn_balltree
__init__(**kwargs)[source]

Methods

__init__(**kwargs)

build(points)

Build the index from a set of points/samples and their coordinate labels.

query(btree, points)

Query points/samples,