In Figure we can see the feature importance profiles of the training and testing data, details of these sets have been highlighted in the supplementary information. Their similarity explains the robust performance over the total H-MoS 2 dataset. In the figure, it is of interest to see that the feature importance of nearest neighbor Sulphur interactions is very high.