zoukankan      html  css  js  c++  java
  • Word2vec 模型载入(tensorflow)

    opts = Options()

    with tf.Graph().as_default(), tf.Session() as session:

    model = Word2Vec(opts, session)

    if FLAGS.interactive:

    #print('load model from file %s %s', opts.save_path, os.path.join(opts.save_path, "/model.ckpt"))

    #model.saver.restore(session, os.path.join(opts.save_path, "/model.ckpt"))

    model.saver.restore(session, "./data/result/model.ckpt")

    #print(model.nearby(['elephant']))

    print('finsh model loading')

    while True:

    print('input your word')

    word = sys.stdin.readline().strip()

    print('haha')

    print(model.nearby(['proton', 'elephant', 'maxwell']))

    print(model.nearby([word]))

       

    [root@cq01-forum-rstree01 embedding]# pwd

    /home/users/chenghuige/other/tensorflow/bazel-bin/tensorflow/models/embedding

    [root@cq01-forum-rstree01 embedding]# ./word2vec_optimized --train_data ./data/text8 --eval_data ./data/questions-words.txt --save_path ./data/result/ --interactive=1

       

       

  • 相关阅读:
    Maven入门指南12:将项目发布到私服
    Groovy学习:第四章 Groovy特性深入
    jQuery部分疑问及小结
    Windows自动化---模拟鼠标键盘
    适配器
    object都有string
    spinner
    context
    OnclickListener
    学习-----领进门,看个人
  • 原文地址:https://www.cnblogs.com/rocketfan/p/5052244.html
Copyright © 2011-2022 走看看