infer.py 620 字节
Newer Older
Z
zhushuang02 已提交
1 2 3 4 5 6 7 8
import paddle.v2 as paddle
import numpy as np

paddle.init(use_gpu=False)
x = paddle.layer.data(name='x', type=paddle.data_type.dense_vector(2))
y_predict = paddle.layer.fc(input=x, size=1, act=paddle.activation.Linear())

# loading the model which generated by training
C
Coke 已提交
9
with open('params_pass_90.tar', 'r') as f:
Z
zhushuang02 已提交
10 11
    parameters = paddle.parameters.Parameters.from_tar(f)

C
Coke 已提交
12 13
# Input multiple sets of data,Output the infer result in a array.
i = [[[1, 2]], [[3, 4]], [[5, 6]]]
Z
zhushuang02 已提交
14
print paddle.infer(output_layer=y_predict, parameters=parameters, input=i)
C
Coke 已提交
15 16 17 18
# Will print:
# [[ -3.24491572]
#  [ -6.94668722]
#  [-10.64845848]]