Added back training the ALL data model as it is used in the live system.