Table 3 Ablation study.
From: Modeling electronic health record data using an end-to-end knowledge-graph-informed topic model
Model | Recon. | Topic quality [ICD,ATC] | |||
|---|---|---|---|---|---|
NLL. | Topic coherence | Topic diversity | Topic quality | TQ (ave.) | |
GAT-ETM | 172.69 | 0.18, 0.314 | 0.76, 0.787 | 0.1368, 0.2471 | 0.1920 |
GAT-ETM (w/o init.) | 179.59 | 0.139, 0.193 | 0.573, 0.447 | 0.0796, 0.0863 | 0.0830 |
GAT-ETM (w/o aug.) | 181.63 | 0.162, 0.282 | 0.733, 0.75 | 0.1187, 0.2115 | 0.1651 |
GETM (w/ aug.) | 180.44 | 0.161, 0.282 | 0.783, 0.807 | 0.1261, 0.2276 | 0.1768 |
