speed it up by training on a random sample of rows