Exemplo n.º 1
0
 def predict_brpc(self, batch_size=1):
     data = np.array([
         0.0137, -0.1136, 0.2553, -0.0692, 0.0582, -0.0727, -0.1583,
         -0.0584, 0.6283, 0.4919, 0.1856, 0.0795, -0.0332
     ]).astype("float32")[np.newaxis, :]
     client = Client()
     client.load_client_config(
         "uci_housing_client/serving_client_conf.prototxt")
     client.connect(["127.0.0.1:9494"])
     fetch_list = client.get_fetch_names()
     fetch_map = client.predict(feed={"x": data},
                                fetch=fetch_list,
                                batch=True)
     return fetch_map
Exemplo n.º 2
0
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# pylint: disable=doc-string-missing

from paddle_serving_client import Client
import sys
import numpy as np

client = Client()
client.load_client_config(sys.argv[1])
client.connect(["127.0.0.1:9393"])
fetch_list = client.get_fetch_names()
import paddle
test_reader = paddle.batch(paddle.reader.shuffle(
    paddle.dataset.uci_housing.test(), buf_size=500),
                           batch_size=1)

for data in test_reader():
    new_data = np.zeros((1, 13)).astype("float32")
    new_data[0] = data[0][0]
    fetch_map = client.predict(feed={"x": new_data},
                               fetch=fetch_list,
                               batch=True)
    print(fetch_map)