Backend: Fix tensorflow_test.go
Signed-off-by: Michael Mayer <michael@liquidbytes.net>
This commit is contained in:
parent
a4b475cbd6
commit
4b7857891a
1 changed files with 1 additions and 1 deletions
|
@ -233,7 +233,7 @@ func TestTensorFlow_BestLabels(t *testing.T) {
|
|||
p[1] = 0.5
|
||||
|
||||
result := tensorFlow.bestLabels(p)
|
||||
assert.Equal(t, "hen", result[0].Name)
|
||||
assert.Equal(t, "chicken", result[0].Name)
|
||||
assert.Equal(t, "bird", result[0].Categories[0])
|
||||
assert.Equal(t, "image", result[0].Source)
|
||||
assert.Equal(t, "goldfish", result[1].Name)
|
||||
|
|
Loading…
Reference in a new issue