|
|
|
@ -131,7 +131,7 @@ class FilmPredictor {
|
|
|
|
|
);
|
|
|
|
|
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);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -140,7 +140,6 @@ class FilmPredictor {
|
|
|
|
|
for(let i = 0; i < arr.length; i++) {
|
|
|
|
|
if(arr[i] < 0.5)
|
|
|
|
|
continue;
|
|
|
|
|
|
|
|
|
|
result.push(genres[i]);
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|