The implementation is heavily inspired from the open source contributions of Xifeng Guo.

This directory contains

1. capsnet ** capsnetv2.py --- Routing with Agreement implementation ** capsulelayers.py-- Capsule layer implementation **capsnetv3.py-- loads saved weights to run tests **tensorflow.sh--runs a shell script for the routing with agreement algo, or loads saved weights

2. convnet ** CNN code

3. class_transition_matrix.py ** dumps confusion matrix, acc, recal

citation

@article{chattopadhyay2020analog, title={Analog forecasting of extreme-causing weather patterns using deep learning}, author={Chattopadhyay, Ashesh and Nabizadeh, Ebrahim and Hassanzadeh, Pedram}, journal={Journal of Advances in Modeling Earth Systems}, volume={12}, number={2}, pages={e2019MS001958}, year={2020}, publisher={Wiley Online Library} }