/
fm_response.go
61 lines (49 loc) · 1.15 KB
/
fm_response.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package eas
import (
"errors"
"fmt"
"github.com/alibaba/pairec/algorithm/response"
)
//easyjson:json
type alinkFMResponseList []*alinkFMResponse
type alinkFMResponse struct {
Result float64 `json:"prediction_result"`
Score float64 `json:"prediction_score"`
//Details string `json:"prediction_detail"`
}
func (r *alinkFMResponse) GetModuleType() bool {
return false
}
func (r *alinkFMResponse) GetScoreMap() map[string]float64 {
var default_val map[string]float64
return default_val
}
func (r *alinkFMResponse) GetScore() float64 {
if r.Result == float64(0) {
return 1 - r.Score
}
return r.Score
}
func alinkFMResponseFunc(data interface{}) (ret []response.AlgoResponse, err error) {
retstr, ok := data.(string)
if !ok {
err = fmt.Errorf("invalid data type, %v", data)
return
}
var result alinkFMResponseList
err = result.UnmarshalJSON([]byte(retstr))
if err != nil {
err = errors.New(fmt.Sprintf("error:%v, body:%s", err, bodyFormat(retstr, 512)))
return
}
for _, res := range result {
ret = append(ret, res)
}
return
}
func bodyFormat(body string, size int) string {
if len(body) <= size {
return body
}
return body[:size]
}