AI Model Random Forest prediction