我已经基于"广泛而深入"的例子创建了一个模型(https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/learn/wide_n_deep_tutorial.py).
我已经将模型导出如下:
m = build_estimator(model_dir) m.fit(input_fn=lambda: input_fn(df_train, True), steps=FLAGS.train_steps) results = m.evaluate(input_fn=lambda: input_fn(df_test, True), steps=1) print('Model statistics:') for key in sorted(results): print("%s: %s" % (key, results[key])) print('Done training!!!') # Export model export_path = sys.argv[-1] print('Exporting trained model to %s' % export_path) m.export( export_path, input_fn=serving_input_fn, use_deprecated_input_fn=False, input_feature_key=INPUT_FEATURE_KEY
我的问题是,如何创建一个客户端来从这个导出的模型进行预测?还有,我是否正确导出了模型?
最终我也需要能够在Java中做到这一点.我怀疑我可以通过使用gRPC从proto文件创建Java类来实现这一点.
文档非常粗略,因此我在这里问.
非常感谢!