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