Learning the trajectory evolution and energy states for BPTI