save pred on first run of mean entropy