Coding a deep learning model using TensorFlow.js