Metadata-Version: 2.1
Name: eclipse-pytorch
Version: 0.0.2
Summary: A pytorch implementation of Eclipse
Home-page: https://github.com/tcapelle/eclipse_pytorch/tree/master/
Author: Thomas Capelle
Author-email: tcapelle@pm.me
License: Apache Software License 2.0
Keywords: pytorch nowcasting solar energy
Platform: UNKNOWN
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3.8
Classifier: License :: OSI Approved :: Apache Software License
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: pip
Requires-Dist: packaging
Requires-Dist: torch (>1.6)
Requires-Dist: fastcore
Provides-Extra: dev

# Eclipse
> Implementing Paletta et al. in Pytorch


![Image](nbs/images/eclipse_diagram.png)

## Install

`pip install eclipse_pytorch`

## How to use

```python
import torch

from eclipse_pytorch.model import *
```

```python
eclipse = Eclipse()
```

```python
preds = eclipse([torch.rand(2, 3, 256, 256) for _ in range(4)])
```

    states.shape=torch.Size([2, 4, 128, 32, 32])
    present_state.shape=torch.Size([2, 1, 128, 32, 32])
    hidden_state.shape=torch.Size([2, 128, 32, 32])
    future_states.shape=torch.Size([2, 6, 128, 32, 32])



