Kneedle (MAE <= 0.416) data pool, yielding a metric of ROC-AUC = 0.6567.LSTM(128) – recurrent layer, tanh activationDropout(0.3)Dense(64, ReLU)Dense(1, sigmoid) – probability of price increaselstm_final_kneedle.keras — the main optimized model.lstm_final_kneedle_cosmetics.keras — model for the alternative dataset (appendix).best_experiments/ — the best models across 14 different algorithms (CatBoost, XGBoost, TabNet, etc.) after pool optimization.| Model | Best Pool | ROC-AUC |
|---|---|---|
| LSTM | Kneedle (MAE ≤ 0.416) | 0.6567 |
| CatBoost | RF Gating (t = 0.7) | 0.6446 |
| XGBoost | ENN (t = 0.7) | 0.6405 |
| TabNet | RF Gating (t = 0.7) | 0.6398 |
| LightGBM | DROP3 (t = 0.7) | 0.6339 |
| Decision Tree | RF Gating (t = 0.7) | 0.6296 |
| Random Forest | RF Gating (t = 0.7) | 0.6294 |
| MLP | ENN (t = 0.7) | 0.6292 |
| Logistic Regression | DT Filter (t = 0.5) | 0.6235 |
| SVM | ENN (t = 0.7) | 0.6178 |
| KNN | Kneedle (MAE ≤ 0.416) | 0.5844 |
| Naive Bayes | DT Filter (t = 0.5) | 0.5731 |
| 1D-CNN | ENN (t = 0.7) | 0.5367 |
| AutoARIMA | DT Filter (t = 0.7) | 0.5027 |
1from huggingface_hub import hf_hub_download
2from tensorflow.keras.models import load_model
3
4# Load the best model
5model_path = hf_hub_download(
6 repo_id="redr1g/final-thesis-experiments",
7 filename="lstm_final_kneedle.keras"
8)
9model = load_model(model_path)