how to get the predicted heading angle

[xiaxx244]

Hi, first I really appreciate your great work. Second, I am wondering where I can obtain the predicted heading angle from the network. Currently, I suspect the phi variable in the dynamic function of https://github.com/StanfordASL/Trajectron-plus-plus/blob/master/trajectron/model/dynamics/unicycle.py should be the value of predicted heading angle. I am wondering if my assumption is correct. Thanks!