vp_suite.models
This modules contains all usable video prediction models.
- MODEL_CLASSES = {'convlstm-shi': <class 'vp_suite.models.precipitation_nowcasting.ef_conv_lstm.EF_ConvLSTM'>, 'copy': <class 'vp_suite.models.copy_last_frame.CopyLastFrame'>, 'lstm': <class 'vp_suite.models.lstm.LSTM'>, 'phy': <class 'vp_suite.models.phydnet.PhyDNet'>, 'predrnn-pp': <class 'vp_suite.models.predrnn_v2.PredRNN_V2'>, 'st-phy': <class 'vp_suite.models.st_phy.STPhy'>, 'trajgru': <class 'vp_suite.models.precipitation_nowcasting.ef_traj_gru.EF_TrajGRU'>, 'unet-3d': <class 'vp_suite.models.unet3d.UNet3D'>}
A dictionary of all models and the corresponding string identifiers with which they can be accessed.