Package to perform Slice Tensor Component Analysis
Project description
SliceTCA
This library provides tools to perform sliceTCA.
Installation
pip install slicetca
Full documentation
The full documentation can be found here.
Examples
Quick example
import slicetca
import torch
from matplotlib import pyplot as plt
device = ('cuda' if torch.cuda.is_available() else 'cpu')
# your_data is a numpy array of shape (trials, neurons, time).
data = torch.tensor(your_data, dtype=torch.float, device=device)
# The tensor is decomposed into 2 trial-, 0 neuron- and 3 time-slicing components.
components, model = slicetca.decompose(data, (2,0,3))
model = slicetca.invariance(model)
slicetca.plot(model)
plt.show()
Notebook
See the example notebook for an application of sliceTCA to publicly available neural data.
Reference
A. Pellegrino@†, H. Stein†, N. A. Cayco-Gaijc@. (2023). Disentangling Mixed Classes of Covariability in Large-Scale Neural Data. https://www.biorxiv.org/content/10.1101/2023.03.01.530616v1.
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
slicetca-0.1.8.tar.gz
(15.8 kB
view hashes)