Sample predict data from classes, Maintain dtype during prediction