未验证 提交 b79fa05b 编写于 作者: D Dong Daxiang 提交者: GitHub

Update IMDB_GO_CLIENT.md

上级 0fa47bfb
...@@ -87,35 +87,35 @@ func main() { ...@@ -87,35 +87,35 @@ func main() {
for { for {
line, err := br.ReadString('\n') line, err := br.ReadString('\n')
if err == io.EOF { if err == io.EOF {
break break
} }
line = strings.Trim(line, "\n") line = strings.Trim(line, "\n")
var words = []int64{} var words = []int64{}
s := strings.Split(line, " ") s := strings.Split(line, " ")
value, err := strconv.Atoi(s[0]) value, err := strconv.Atoi(s[0])
var feed_int_map map[string][]int64 var feed_int_map map[string][]int64
for _, v := range s[1:value + 1] { for _, v := range s[1:value + 1] {
int_v, _ := strconv.Atoi(v) int_v, _ := strconv.Atoi(v)
words = append(words, int64(int_v)) words = append(words, int64(int_v))
} }
label, err := strconv.Atoi(s[len(s)-1]) label, err := strconv.Atoi(s[len(s)-1])
if err != nil { if err != nil {
panic(err) panic(err)
} }
feed_int_map = map[string][]int64{} feed_int_map = map[string][]int64{}
feed_int_map["words"] = words feed_int_map["words"] = words
feed_int_map["label"] = []int64{int64(label)} feed_int_map["label"] = []int64{int64(label)}
result = serving_client.Predict(handle, feed_int_map, fetch) result = serving_client.Predict(handle, feed_int_map, fetch)
fmt.Println(result["prediction"][1], "\t", int64(label)) fmt.Println(result["prediction"][1], "\t", int64(label))
} }
} }
``` ```
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册