A method for improving a lithographic process for imaging a portion of a design layout onto a substrate using a lithographic apparatus, the method including: obtaining a first source of the lithographic apparatus; classifying the first source into a class among a plurality of possible classes, based on one or more numerical characteristics of the first source, using a machine learning model, by a computer; determining whether the class is among one or more predetermined classes; only when the class is among the one or more predetermined classes, adjusting one or more source design variables to obtain a second source.