Added API methods for training related operations