Code accompanying publication https://doi.org/10.1176/appi.ajp.2021.21010095 - View it on GitHub