Java调用Tensorflow训练模型预测结果

黄然大悟

于 2018-12-28 21:41:51 发布

阅读量5.6k

点赞数 2

分类专栏：自然语言处理 Tensorflow & Keras 文章标签： Tensorflow Java调用Tensorflow模型

本文链接：https://blog.youkuaiyun.com/huanghaocs/article/details/85310256

版权

这篇博客介绍了如何使用Java调用已训练的Tensorflow模型进行预测。首先，需要读取词典文件和加载模型，接着读取预测数据，最后执行预测并获取结果。在Java中，通过创建Tensorflow Session并保持输入输出格式的一致性来实现模型预测。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Java调用Tensorflow训练好的模型做预测，首先需要读取词典，然后加载模型，读入数据，最后预测结果。
模型训练参考上一篇博客：使用Tensorflow训练LSTM+Attention中文标题党分类

首先需要下载一些包，如果是maven项目在pom.xml中添加两个依赖。

<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>tensorflow</artifactId>
    <version>1.5.0</version>
</dependency>
<dependency>
    <groupId>org.tensorflow</groupId>
    <artifactId>libtensorflow_jni</artifactId>
    <version>1.5.0</version>
</dependency>

读取词典文件

这个词典文件wordIndexMap.txt，就是上一篇对应训练模型之前生成的词典文件。每行一个词和词的编号。

// 从文件读取词典文件存入Map
private static Map<String, Integer> readVocabFromFile(String pathname) throws IOException{
   
    Map<String, Integer> wordMap = new HashMap<String, Integer>();
    File filename = new File(pathname);
    InputStreamReader reader = new InputStreamReader(new FileInputStream(filename));
    BufferedReader br = new BufferedReader(reader);
    String line = "";
    line = br.readLine();
    String[] lineArray;
    while(line != null){
   
        lineArray = line.split(" "

最低0.47元/天解锁文章