ParametricSpectralClustering 0.0.1

A library for users to use parametric spectral clustering

Parametric Spectral Clustering

This repository provides a PyTorch implementation of the Parametric Spectral Clustering (PSC) algorithm, which offers a favorable alternative to the traditional spectral clustering algorithm. PSC addresses issues related to computational efficiency, memory usage, and the absence of online learning capabilities. It serves as a versatile framework suitable for applying spectral clustering to large datasets.

Installation

Dependencies

Parametric Spectral Clustering requires:

• Python (>= 3.8)
• NumPy (>= 1.24.3)
• SciPy (>= 1.10.3)
• PyTorch (>= 1.12.1)
• scikit-learn (>= 1.3.2)
• Pandas (>= 2.0.3)

---

User installation

Use setup.py:

python setup.py install

Use pip:

pip install -i [url] ParametricSpectralClustering==0.0.14

Sample Usage

Using UCI ML hand-written digits datasets as an example.

>>> from ParametricSpectralClustering import PSC, Four_layer_FNN
>>> from sklearn.datasets import load_digits
>>> from sklearn.cluster import KMeans
>>> digits = load_digits()
>>> X = digits.data/16
>>> cluster_method = KMeans(n_clusters=10, init="k-means++", n_init=1, max_iter=100, algorithm='elkan')
>>> model = Four_layer_FNN(64, 128, 256, 64, 10)
>>> psc = PSC(model=model, clustering_method=cluster_method, n_neighbor=10, sampling_ratio=0, batch_size_data=1797)
>>> psc.fit(X)
>>> psc.save_model("model")
>>> cluster_idx = psc.predict(X)

Command line tool

After installation, you may run the following scripts directly.

python bin\run.py [data] [rate] [n_cluster] [model_path] [cluster_result_format]

The [data] can accept .txt, .csv, and .npy format of data.

The [rate] should be in float, between 0.0 and 1.0. It represent the proportion of the input data reserved for training the mapping function from the original feature space to the spectral embedding.

The [n_cluster] is the number of clusters the user intends to partition. This number needs to be lower than the total data points available within the dataset.

The [model_path] is the path to save the trained model.

The [cluster_result_format] can be either .txt or .csv. It represent the format of the cluster result.

License

Distributed under the MIT License. See LICENSE.txt for more information.

Contact

Author	Ivy Chang	Hsin Ju Tai
E-mail	[e-mail]	[e-mail]

Project Link: TBA

Change Log

0.0.10 (2023/08/21)

• First release

0.0.13 (2023/08/21)

• Fix pickle