Classify News with CNN Usage Train the model by running python train. It will run 20 epochs and save the model as pb format. In predict.py, there is serving code where you can test a news prediction.