Spectral Bridges clustering algorithm
Project description
Spectral Bridges
Spectral Bridges is a Python package that implements a novel clustering algorithm combining k-means and spectral clustering techniques. It leverages efficient affinity matrix computation and merges clusters based on a connectivity measure inspired by SVM's margin concept. This package is designed to provide robust clustering solutions, particularly suited for large datasets.
Features
- Spectral Bridges Algorithm: Integrates k-means and spectral clustering with efficient affinity matrix calculation for improved clustering results.
- Scalability: Designed to handle large datasets by optimizing cluster formation through advanced affinity matrix computations.
- Customizable: Parameters such as number of clusters, iterations, and random state allow flexibility in clustering configurations.
Speed
Starting with version 1.0.0, Spectral Bridges not only utilizes FAISS's efficient k-means implementation but also employs SimSIMD for centroid initialization isntead of scikit-learn. This combination results in a small speed improvement.
Installation
You can install the package via pip:
pip install spectral-bridges
Usage
Example
from spectralbridges import SpectralBridges
import numpy as np
# Generate sample data
np.random.seed(0)
X = np.random.rand(100, 10) # Replace with your dataset
# Initialize and fit Spectral Bridges
model = SpectralBridges(n_clusters=5, n_nodes=10, random_state=42)
model.fit(X)
# Predict clusters for new data points
new_data = np.random.rand(20, 10) # Replace with new data
predicted_clusters = model.predict(new_data)
print("Predicted clusters:", predicted_clusters)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for spectral_bridges-1.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8af40d95fc00e65decddffac9ca75451ba145c4b90b768a07a8db01f8a94e126 |
|
MD5 | 123dd5561575bc64048629e45c04b1fb |
|
BLAKE2b-256 | 141fe661d4511d3645040f19c982d45db756c9e9d309270befd09f2f81885f7b |