Table 7 Comparison of models’ training time and memory usage.
Model | Multi label data set | Binary label data set | ||
---|---|---|---|---|
Training time | Memory used | Training time | Memory used | |
RF | 17 Seconds | 171 MB | 13 Second | 123 MB |
KNN | 3 Minutes | 291 MB | 2.3 Minutes | 197 MB |
LSTM | 9.4 Minutes | 731 MB | 7.3 Minutes | 983 MB |
RF- LSTM Stacked Tune KNN | 14 Minutes | 1.3 GB |