小编Jan*_*ers的帖子

如何在Java中使用TensorFlow LinearClassifier

在Python中,我训练了TensorFlow LinearClassifier并将其保存为:

model = tf.contrib.learn.LinearClassifier(feature_columns=columns)
model.fit(input_fn=train_input_fn, steps=100)
model.export_savedmodel(export_dir, parsing_serving_input_fn)
Run Code Online (Sandbox Code Playgroud)

通过使用TensorFlow Java API,我可以使用以下方法在Java中加载此模型:

model = SavedModelBundle.load(export_dir, "serve");
Run Code Online (Sandbox Code Playgroud)

看来我应该能够使用类似

model.session().runner().feed(???, ???).fetch(???, ???).run()
Run Code Online (Sandbox Code Playgroud)

但是我应该从图形中获取/获取哪些变量名称/数据以提供其功能并获取类的概率?据我所知,Java文档缺少此信息。

java tensorflow

1
推荐指数
1
解决办法
1178
查看次数

标签 统计

java ×1

tensorflow ×1