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

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}_native/fold_$FOLD
SHAP_DIR=contributions/${MODEL_NAME}_native/fold_$FOLD
INPUT_DATA=input_data/${MODEL_NAME}_native/input_data_predict.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