Figure 9. Comparison of profiles predicted by the model (shaded region represents $\pm$ one standard deviation of the Monte Carlo dropout ensemble) to the expected profiles for shot 204105. This shot was in the testing dataset (not used in training or topology selection).