Guest User

Untitled

a guest
Jan 16th, 2018
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.47 KB | None | 0 0
  1. res := getTopFiveLabels(labels, output[0].Value().([][]float32)[0])
  2. for _, l := range res {
  3. fmt.Printf("label: %s, probability: %.2f%%\n", l.Label, l.Probability*100)
  4. }
  5.  
  6. func getTopFiveLabels(labels []string, probabilities []float32) []Label {
  7. var resultLabels []Label
  8. for i, p := range probabilities {
  9. if i >= len(labels) {
  10. break
  11. }
  12. resultLabels = append(resultLabels, Label{Label: labels[i], Probability: p})
  13. }
  14.  
  15. sort.Sort(Labels(resultLabels))
  16. return resultLabels[:5]
  17. }
Add Comment
Please, Sign In to add comment