mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-12 23:12:39 +08:00
Dr. Kashif Rasul
33 lines
660 B
Markdown
33 lines
660 B
Markdown
# PyTorch-TS
|
|
|
|
PyTorch-TS is a PyTorch Probabilistic Time Series Modeling framework which provides state of the art time series models and utilities for loading and iterating over time series datasets.
|
|
|
|
## Installation
|
|
|
|
```
|
|
$ pip3 install pytorch-ts
|
|
```
|
|
|
|
## Usage
|
|
|
|
```python
|
|
import pandas as pd
|
|
|
|
from pts.dataset import ListDataset
|
|
|
|
url = "https://raw.githubusercontent.com/numenta/NAB/master/data/realTweets/Twitter_volume_AAPL.csv"
|
|
df = pd.read_csv(url, header=0, index_col=0)
|
|
|
|
training_data = ListDataset(
|
|
[{"start": df.index[0], "target": df.value[:"2015-04-05 00:00:00"]}],
|
|
freq = "5min"
|
|
)
|
|
```
|
|
|
|
## Development
|
|
|
|
```
|
|
pip install -e .
|
|
pytest test
|
|
```
|