Forestcluster

Python package for clustering assisted by the random forest.

Installation

$ python -m pip install git+https://github.com/nbsato/forestcluster.git

Usage

Assume that you have a fitted sklearn.ensemble.RandomForestRegressor object, estimator, and a set of feature vectors, X.

Import the package:

>>> import forestcluster

Transform the feature vectors:

>>> Xt = forestcluster.transform(estimator, X)

Perform clustering:

>>> Z = forestcluster.cluster(Xt)

Assign cluster labels to samples:

>>> labels = forestcluster.assign_labels(Z)

License

Apache License, Version 2.0