implemented a model for handling the random prediction generations