Demos/film-poster-genres
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

FilmPredictor: Fixed prediction error

Browse source
This commit is contained in:
Starbeamrainbowlabs 2020-09-15 19:18:20 +01:00
parent 304aa66358
commit c9e7826416
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/lib/FilmPredictor.mjs
View file

@ -131,7 +131,7 @@ class FilmPredictor {
); );
return image.reshape([1, ...image.shape ]); return image.reshape([1, ...image.shape ]);
}) })
let result_array = await this.model.predict(imagetensor).array(); let result_array = (await this.model.predict(imagetensor).array())[0];
return this.array2genres(result_array); return this.array2genres(result_array);
} }
@ -140,7 +140,6 @@ class FilmPredictor {
for(let i = 0; i < arr.length; i++) { for(let i = 0; i < arr.length; i++) {
if(arr[i] < 0.5) if(arr[i] < 0.5)
continue; continue;
result.push(genres[i]); result.push(genres[i]);
} }
return result; return result;