This is just a simple model and a very small dataset, I used this as example just to show you how to do things. if you want good results you need to do way more than this model does. Try bigger problems.