Skip to content

A small library for simplifying the training process of a deep learning time series classification task using a PyTorch backend, and provide prediction explanations using the Captum library capabilities.

Notifications You must be signed in to change notification settings

rabdiego/captime

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Captime

A small library for simplifying the training process of a deep learning time series classification task using a PyTorch backend, and provide prediction explanations using the Captum library capabilities.

Classification

You can train, evaluate and deploy deep learning models using a simple sklearn-like interface with the classifiers provided in this section:

  • MLPClassifier
  • CNNClassifier
  • GRUClassifier
  • LSTMClassifier

Currently, all classifiers only support the Pandas DataFrame format for time series data.

Dataset

A small dataset class used for converting DataFrames into Torch Datasets.

Explainers

Using TSExplainer's .interval_plot() you can see which time intervals contributed the most for your model final prediction, with support for multi-dimensional series.

example

About

A small library for simplifying the training process of a deep learning time series classification task using a PyTorch backend, and provide prediction explanations using the Captum library capabilities.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published