在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文档缺少此信息。