Skip to main content
. 2021 Nov 22;11(11):1281. doi: 10.3390/life11111281
Algorithm 6. The AI-Score ensemble description.
Input: SCB training dataset, SCB testing dataset.
Output: prediction probabilities for each diagnosis class (Mild, Moderate, Severe)
1. Select a 5-fold split of the train SCB Dataset.
2. base_models = [“Ada Boost”, “Random Forests”,” XGBoost”]
3. meta_model = “CatBoost”
4. For each model in base_models:
5. Evaluate the model using 5-fold cross-validation.
6. Save all out-of-fold predictions.
7. Fit the model on the full training dataset and save.
8. Fit the meta-model on the out-of-fold predictions from the previous layer.
9. Evaluate the model on the SCB testing dataset.
10. For each class in the set of diagnoses do
11.  output prediction probability