Tag: tensorflow serving

用Java加载sklearn模型。 在python中使用DNNClassifier创建的模型

目标是在Java中使用tensorflow.contrib.learn.learn.DNNClassifier在python中创建/训练的模型。 目前主要问题是要知道在会话运行方法上用java给出的“张量”的名称。 我在python中有这个测试代码: from __future__ import division, print_function, absolute_import import tensorflow as tf import pandas as pd import tensorflow.contrib.learn as learn import numpy as np from sklearn import metrics from sklearn.cross_validation import train_test_split from tensorflow.contrib import layers from tensorflow.contrib.learn.python.learn.utils import input_fn_utils from tensorflow.python.ops import array_ops from tensorflow.python.framework import dtypes from tensorflow.python.util.compat import as_text print(tf.VERSION) df […]

使用估算器在java中加载/提供tensorflow模型时出现问题

我使用了人口普查数据,并使用张量流中的估计量api创建了一个广泛而深入的模型。 在Java中加载模型时,似乎存在一个错误,它不允许加载模型。 看起来像是例外 Exception in thread “main” org.tensorflow.TensorFlowException: Op type not registered ‘SparseFeatureCross’ in binary running on gmalhotra-mba-2.local. Make sure the Op and Kernel are registered in the binary running in this process. at org.tensorflow.SavedModelBundle.load(Native Method) at org.tensorflow.SavedModelBundle.load(SavedModelBundle.java:39) at deeplearning.DeepLearningTest.main(DeepLearningTest.java:32) 请找到以下用于保存模型的python代码: https : //gist.github.com/gaganmalhotra/cd6a5898b9caf9005a05c8831a9b9153 使用的Java代码如下: public static void main(String[] args) { try (SavedModelBundle b = […]