Machine learning prediction models for AFM virtual imaging system