1234567891011121314151617181920212223242526272829303132 |
- '''
- 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
|