compare.py 589 B

1234567891011121314151617181920212223242526272829303132
  1. '''
  2. Created on 2019年6月13日
  3. @author: User
  4. '''
  5. import fool
  6. from bi_lstm_crf import *
  7. bilstm = BiLSTM()
  8. ''''''
  9. saver = tf.train.Saver()
  10. path_add = "0-11/"
  11. saver.restore(bilstm.sess, 'model/'+path_add+'model.ckpt')
  12. def compare(text):
  13. print(fool.ner(text))
  14. '''
  15. bilstm.initVariables()
  16. '''
  17. #init_op = tf.global_variables_initializer()
  18. #sess.run(init_op)
  19. #summaryWriter = tf.summary.FileWriter('log/', tf.get_default_graph())
  20. print(bilstm.ner(text))
  21. if __name__=="__main__":
  22. text = '白云区物价局价格认证中心'
  23. compare(text)
  24. pass