Abstract: In example embodiments, techniques are provided for labeling elements of an infrastructure model with classes. The techniques may be implemented by a labeling tool that uses an ML model to create element selections and provides a cycle review mode to speed review within such selections. The labeling tool may further provide for two file loading and a number of visualization schemes to speed comparison of label files and prediction files.