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
PREDICTIONS_DIR=predictions/$MODEL_NAME/fold_$FOLD
INPUT_DATA=$DATA_DIR/input_data_fusion_deconvolution_V3.json

mkdir -p $PREDICTIONS_DIR
bpnet-predict \
    --model ../models/$MODEL_NAME/fold_$FOLD/model_split000 \
    --chrom-sizes $CHROM_SIZES \
    --chroms chr1 chr2 chrX chr3 chr4 chr5 chr6 chr7 chr10 chr8 chr14 chr9 chr11 chr13 chr12 chr15 chr16 chr17 chrY chr18 chr19 chr20 chr21 chr22 chrM\
    --test-indices-file None \
    --reference-genome $REFERENCE_GENOME \
    --output-dir $PREDICTIONS_DIR \
    --input-data $INPUT_DATA \
    --sequence-generator-name BPNet \
    --input-seq-len 2114 \
    --output-len 1000 \
    --output-window-size 1000 \
    --batch-size 64 \
    --reverse-complement-average \
    --threads 2 \
    --generate-predicted-profile-bigWigs