From: Shapley-based interpretation of deep learning models for wildfire spread rate prediction
Tool/technique | Purpose | Version |
---|---|---|
TensorFlow/Keras | Model building and training | TensorFlow 2.0 / Keras 2.3.1 |
PyTorch | Alternative model building and training framework | PyTorch 1.7.1 |
Scikit-learn | Preprocessing and model evaluation | Scikit-learn 0.23.2 |
SHAP | Interpreting model predictions | SHAP 0.36.0 |