Почему мой код Tensorflow. js работает так медленно? - PullRequest
0 голосов
/ 21 марта 2020

Я пытаюсь создать AI Flappy Bird с нейроэволюцией, используя Tensorflow. js и моя функция прогнозирования сильно замедляет программу. Я сделал что-то не так?

async predict(inputs) {

   //convert the inputs array to a tensor
   let xs = tf.tensor2d([inputs]);

   //predict  the outputs
   let ys = this.model.predict(xs);

   // convert the outputs to an array
   let outputs = await ys.data();

   xs.dispose();
   ys.dispose();
   return outputs;
}

Также я должен упомянуть, что модель является последовательной моделью.

...