Skip to content
/ pcax Public

Minimal Principal Component Analysis (PCA) implementation in JAX

License

Notifications You must be signed in to change notification settings

alonfnt/pcax

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PCAx: Minimal Principal Component Analysis in Jax.

The aim of this project is to provide a JAX-based PCA implementation, eliminating the need for unnecessary data transfer to CPU or conversions to Numpy. This can provide performance benefits when working with large datasets or in GPU-intensive workflow

Usage

import pcax

# Fit the PCA model with 3 components on your data X
state = pcax.fit(X, n_components=3)

# Transform X to its principal components
X_pca = pcax.transform(state, X)

# Recover the original X from its principal components
X_recover = pcax.recover(state, X_pca)

Installation

pcax can be installed from PyPI via pip

pip install pcax

Alternatively, it can be installed directly from the GitHub repository:

pip install git+git://github.com/alonfnt/pcax.git

About

Minimal Principal Component Analysis (PCA) implementation in JAX

Topics

Resources

License

Stars

Watchers

Forks

Languages