(is_predict)
| 61 | self.wfile.write(bytes(str(group_id), "utf-8")) |
| 62 | |
| 63 | def main(is_predict): |
| 64 | if is_predict: |
| 65 | saver.restore(sess, 'data/model/model') |
| 66 | #predict(samples, sess, x, y, y_, train_step) |
| 67 | myServer = HTTPServer(("0.0.0.0", 5001), MyServer) |
| 68 | print("begin listen") |
| 69 | myServer.serve_forever() |
| 70 | else: |
| 71 | train(samples, sess, x, y, y_, train_step) |
| 72 | |
| 73 | |
| 74 | if __name__ == '__main__': |