[ebf7be]: / code / main.py

Download this file

24 lines (14 with data), 396 Bytes

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