Methods and systems are described for training a machine learning (ML) model to predict the gain of a target channel of a multi-channel amplifier device. An ML model may be pre-trained using an existing set of training objects. The trained ML model then can be utilized to suggest further useful training objects to be labelled that will improve the performance of the ML model by predicting more accurate target channel gains given the on/off value for the channel inputs.