Skip to content

Machine Learning: Model Architecture

qedk edited this page Nov 13, 2020 · 1 revision

Architecture diagram

The model has over 13,738,513 parameters (4,689,013 trainable + 9,049,500 untrainable) and will take a significant time to train without a TPU/GPU. The model tries to strike a balance between efficiency and performance, and while training is slow, mood analysis of an average paragraph takes <1 second on a 2017 Core i5 processor.