Demo model trained for 1 epoch on 4096 examples from the yelp_polarity dataset.