A lithology prediction that uses a geological age model as an input to a machine learning model. The geological age model is capable of separating and recoding different seismic packages derived from the horizon interpretation. Once the machine learning model has been trained, a validation may be performed to determine the quality of the machine learning model. The quality may be improved by refining the training of the machine learning model. The lithology prediction generated by the machine learning model that utilizes the geological age model provides an improved lithology prediction that more accurately reflects the subterranean formation of an area of interest.