Abstract: A method of training a system for making predictions relating to products manufactured via a manufacturing process includes receiving a plurality of input vectors and a plurality of defect values corresponding to the plurality of input vectors, identifying a plurality of first cluster labels corresponding to the plurality of input vectors based on the defect values, training a cluster classifier based on the input vectors and the corresponding first cluster labels, reassigning the input vectors to a plurality of second cluster labels based on outputs of the cluster classifier, retraining the cluster classifier based on the input vectors and the second cluster labels, and training a plurality of machine learning models corresponding to the second cluster labels.