Table 3 Evaluation metrics used to assess the performance and effectiveness of various regression models trained on two datasets for predicting Curie temperature
From: The northeast materials database for magnetic materials
Model | Dataset | R-squared (Test) | MAE (Test) | RMSE (Test) |
|---|---|---|---|---|
Random Forest | Original | 0.82 | 71K | 115K |
Balanced (Undersampled) | 0.85 | 65K | 104K | |
Ensembled Neural Network | Original | 0.84 | 60K | 110K |
Balanced (Undersampled) | 0.85 | 56K | 103K | |
XGBoost | Original | 0.84 | 62K | 107K |
Balanced (Undersampled) | 0.87 | 56K | 97K |