DATA_DIR=$1
MODEL_NAME=$2
FOLD=$3

GPU=0
export CUDA_VISIBLE_DEVICES=$GPU

REFERENCE_DIR=../data
CHROM_SIZES=$REFERENCE_DIR/hg38.chrom.sizes
REFERENCE_GENOME=$REFERENCE_DIR/hg38.genome.fa
MODEL_DIR=../models/$MODEL_NAME/fold_$FOLD
SHAP_DIR=contributions/$MODEL_NAME/fold_$FOLD
INPUT_DATA=$DATA_DIR/input_data_fusion_deconvolution_V3.json

mkdir -p $SHAP_DIR
bpnet-shap \
        --reference-genome $REFERENCE_GENOME \
        --model $MODEL_DIR/model_split000  \
        --bed-file $DATA_DIR/idr_peaks/peaks_inliers.bed \
        --output-dir $SHAP_DIR \
        --input-seq-len 2114 \
        --control-len 1000 \
        --task-id 0 \
        --input-data $INPUT_DATA \
        --chrom-sizes $CHROM_SIZES \
        --generate-shap-bigWigs