Skip to content

Add UMAP support for dimension reduction - #588

Merged
aatmanvaidya merged 6 commits into
tattle-made:developmentfrom
SarthakB11:feature/add-umap-support
Apr 20, 2025
Merged

aatmanvaidya merged 6 commits into
tattle-made:developmentfrom
SarthakB11:feature/add-umap-support

Conversation

@SarthakB11

Copy link
Copy Markdown
Contributor

Overview

PR under Issue #574 - Add UMAP support to Dimension Reduction Operator
I've added UMAP (Uniform Manifold Approximation and Projection) support to the dimension reduction operator. UMAP is a dimensionality reduction technique that can be used as an alternative to t-SNE, offering similar visualization quality with better preservation of global structure and faster performance on larger datasets.

Changes Made

  • Added UMAPReduction implementation to the dimension reduction operator
  • Updated the DimensionReductionFactory to support creating UMAP models
  • Added umap-learn dependency to pyproject.toml and requirements
  • Added tests for UMAP functionality
  • Fixed import sorting issues

Testing

  • Added unit tests specifically for UMAP reduction
  • Ensured existing t-SNE functionality continues to work
  • Verified that the dimension reduction operator properly handles both methods

These changes allow users to choose between t-SNE and UMAP for their dimension reduction needs, providing more flexibility and potentially better performance for large datasets.

@aatmanvaidya
aatmanvaidya self-requested a review April 19, 2025 15:59
@aatmanvaidya

Copy link
Copy Markdown
Collaborator

hi @SarthakB11 just the changes in the operators/dimension_reduction/ folder are enough, no need to make any changes in the src/core/operators/ folder

@aatmanvaidya
aatmanvaidya merged commit 3c245b3 into tattle-made:development Apr 20, 2025
@aatmanvaidya aatmanvaidya linked an issue Apr 20, 2025 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add UMAP support to Dimension Reduction Operator

2 participants