'''
Created on 2019年6月13日

@author: User
'''


import fool
from bi_lstm_crf import *

bilstm = BiLSTM()

''''''
saver = tf.train.Saver()
path_add = "0-11/"
saver.restore(bilstm.sess, 'model/'+path_add+'model.ckpt')

def compare(text):
    print(fool.ner(text))
    '''
    bilstm.initVariables()
    '''
    #init_op = tf.global_variables_initializer()
    #sess.run(init_op)
    #summaryWriter = tf.summary.FileWriter('log/', tf.get_default_graph())
    print(bilstm.ner(text))

if __name__=="__main__":
    text = '白云区物价局价格认证中心'
    compare(text)
    pass