a | b/code/main.py | ||
---|---|---|---|
1 | # -*- coding:utf-8 -*- |
||
2 | """ |
||
3 | Start entry for training and testing |
||
4 | """ |
||
5 | import tensorflow as tf |
||
6 | import os |
||
7 | from model2 import model |
||
8 | import numpy as np |
||
9 | |||
10 | from project_config import * |
||
11 | |||
12 | if __name__ =='__main__': |
||
13 | print(" beigin...") |
||
14 | epoch = 5 |
||
15 | model = model(learning_rate,keep_prob,batch_size,epoch) |
||
16 | model.inference(normalization_output_path,test_path,2,True) |
||
17 | |||
18 | |||
19 | |||
20 | |||
21 | |||
22 | |||
23 |