Automated, predictive, and interpretable inference of C. elegans escape dynamics