#clustering #cluster #numbers #python-bindings #predefined #algorithm #mean-shift

meanshift-rs

Rust版本的MeanShift聚类算法,无需预定义簇的数量

6个版本

0.9.3 2022年10月25日
0.9.2 2022年10月25日
0.8.0 2022年6月7日
0.7.0 2022年6月3日

8#predefined 中排名

每月下载量 37
用于 s2gpp

MIT 协议

49KB
1K SLoC

MeanShift-rs

pipeline status coverage report

Git tag

作为Python包安装

0. 最小要求

  • rustc1.54.0
  • Python3.7

1. 创建Python环境

python3 -m venv .venv

2. 安装MeanShift-rs

make install

pip install -r requirements.txt
bash ./tasks.sh release-install

pip install meanshift-rs

用法

Rust

use meanshift_rs::{MeanShiftActor, Parameters};

fn example_fn(dataset: Array2<f64>) {
    let parameters = Parameters::default();
    let mut mean_shift = MeanShiftActor::init(parameters);
    let (centers, labels) = mean_shift.fit(dataset)
        .expect("No MeanShiftResult was returned!");
}

Python

from meanshift_rs import MeanShift

# ... load data

ms = MeanShift()
ms.fit(data)

print(ms.cluster_centers)
print(ms.labels)

依赖

~6–16MB
~193K SLoC