How to replicate Fig. 1

[kuchaale]

How shall I replicate Fig. 1 from your paper for educational purposes? Is

SHORT/hm_model.py

Line 1830 in 1631975

def unit_test():

a place to start? Thanks for your help.

[justinfocus12]

Thanks for your interest!

To clarify, are you referring to the paper "Learning Forecasts of Rare Stratospheric Transitions from Short Simulations" (https://journals.ametsoc.org/view/journals/mwre/149/11/MWR-D-21-0024.1.xml)? If so, do you want to reproduce all parts of the figure or just a subset?

The function you've identified is just a check of correctness. I refer you instead to the driver file hm_forecast_driver.py, specifically

SHORT/hm_forecast_driver.py

Line 95 in 1631975

fig,ax = model.plot_two_snapshots(model.xst[0],model.xst[1],asymb,bsymb)

which calls the function hm_model.plot_two_snapshots

SHORT/hm_model.py

Line 1566 in 1631975

def plot_two_snapshots(self,xt0,xt1,suffix0="",suffix1=""):

.

This will create a plot that shows (a) radiative cooling profile, (b) zonal wind profiles, and (c) streamfunction contours, for both of the provided initial conditions in the same plot.

File paths will have to be modified of course for your machine, and you can greatly reduce the integration lengths just to find fixed points.