Upgrade prediction api sample code to v 1.6 and fix training data.