This is a simplified version of PyRain https://github.com/FrontierDevelopmentLab/PyRain. The platform is collab.
import tes.ipynb
This file is used to figure out how to use dataset function
model test.ipynb
This file contains the custom ConvLSTM model builded according to the author's codes and used to test if it can run correctly.