Please check author’s (Cynthia Wu) webpage for original code and documentation: http://faculty.chicagobooth.edu/jing.wu/research/zip/WX.zip
Only the main file (main_srtsm.m) in the original code has been replicated in python along with the related dependencies. The script generates 6 plots and the shadow rate in a csv format.