Skip to content

Latest commit

 

History

History
28 lines (23 loc) · 712 Bytes

README.md

File metadata and controls

28 lines (23 loc) · 712 Bytes

ROCKET

A prototype implementation for ROCKET - Exceptionally fast and accurate time series classification using random convolutional kernels https://arxiv.org/abs/1910.13051

Installation

Build an environment:

virtualenv venv
source venv/bin/activate
pip3 install -r requirements

Training

In order to train the model, run:

python3 train.py

Testing

Testing the trained can be done by running:

python3 test.py

The script will choose the best model from the trained models folder, will sample a series from the dataloader, predict the class and plot it

Prediction Results:

Example prediction plot1

Example prediction plot1