Initial check in of train predicion API data sampling script.