Skip to main content

Table 6 Tools and techniques

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