yjxiong/clustering-benchmark

toolkit for benchmark clustering algorithms

clustering-benchmark

toolkit for benchmark clustering algorithms

Install

Run

python setup.py install

Usage

After installed, use it as a library

from clustering_benchmark import ClusteringBenchmark

bm = ClusteringBenchmark(gt_data=gt)

scores = bm.evaluate_vmeasure(prediction)

print("Evaluation results:")
print(scores)

For details, please see the script in tools/run_clustering_test.py