FilesExpand file tree

 master

/

example.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

21 lines (14 loc) · 657 Bytes

 master

/

example.py

Copy path

Top

File metadata and controls

• Code
• Blame

21 lines (14 loc) · 657 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

import logging

from predict_client.prod_client import ProdClient

logging.basicConfig(level=logging.DEBUG,

format='%(asctime)s - %(levelname)s - %(name)s - %(message)s')

# In each file/module, do this to get the module name in the logs

logger = logging.getLogger(__name__)

# Make sure you have a model running on localhost:9000

host = 'localhost:9000'

model_name = 'simple'

model_version = 1

client = ProdClient(host, model_name, model_version)

req_data = [{'in_tensor_name': 'a', 'in_tensor_dtype': 'DT_INT32', 'data': 2}]

prediction = client.predict(req_data, request_timeout=10)

logger.info('Prediction: {}'.format(prediction))