prediction_dir is defined but not used in the notebook.
prediction_dir is defined but not used in the notebook.