UMAP-clustering

This repository contains a UMAP-based product clustering script.

Files

• UMAP.py: Performs UMAP dimensionality reduction and KMeans clustering on product cooccurrence data.
• product_cooccurrence.csv: Product cooccurrence matrix.
• products.csv: Product metadata.

Usage

Install requirements:

pip install numpy pandas scikit-learn umap-learn matplotlib

Run:

python UMAP.py