Skip to content

Commit

Permalink
lstm: fix dataset and plots
Browse files Browse the repository at this point in the history
  • Loading branch information
thatgeeman committed May 19, 2024
1 parent a52af62 commit cb16a2d
Show file tree
Hide file tree
Showing 14 changed files with 1,133 additions and 950 deletions.
2 changes: 1 addition & 1 deletion Pipfile.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

39 changes: 22 additions & 17 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,32 +1,37 @@
ts_vae_lstm
================


<!-- WARNING: THIS FILE WAS AUTOGENERATED! DO NOT EDIT! -->

This file will become your README and also the index of your
documentation.
## TS VAE-LSTM

## Install
> Implementation of the paper [Anomaly Detection for Time Series Using
> VAE-LSTM Hybrid Model](https://ieeexplore.ieee.org/document/9053558)
``` sh
This is a work in progress.

### TODO

- [ ] Separate training from notebooks
- [ ] use a squared term if the absolute element-wise error falls below
delta and a delta-scaled L1 term otherwise (Huber)
- [x] Use `.project-env` to manage paths

``` sh
pip install ts_vae_lstm
```

## CUDA
### Env variables

Download the driver and cuda version compiled for the driver.
```bash
sudo mhwd -i pci video-nvidia-470xx
sudo pacman -U https://archive.archlinux.org/packages/c/cuda/cuda-11.4.2-1-x86_64.pkg.tar.zst
``` bash
BASEDIR='<your-base-path>/ts_vae-lstm'
MODELDIR=${BASEDIR}/models
```

## How to use
### CUDA

Fill me in please! Don’t forget code examples:
Download the driver and cuda version compiled for the driver.

``` python
1+1
``` bash
sudo mhwd -i pci video-nvidia-470xx
sudo pacman -U https://archive.archlinux.org/packages/c/cuda/cuda-11.4.2-1-x86_64.pkg.tar.zst
```

2
Loading

0 comments on commit cb16a2d

Please sign in to comment.