#!/bin/bash
#SBATCH -p akundaje,owners
#SBATCH -t 02:00:00
#SBATCH -c 32
#SBATCH --mem=64G
#SBATCH --requeue

# Initialize conda for this non-interactive shell
eval "$(conda shell.bash hook)"
conda activate chrombpnet
module load system
module load cairo

MODEL_NAME=$1

SHAP_DIR=contributions/$MODEL_NAME/with_bias
MODISCO_DIR=modisco/$MODEL_NAME/with_bias
MODISCO_COUNTS_DIR=$MODISCO_DIR/counts
MODISCO_PROFILE_DIR=$MODISCO_DIR/profile
mkdir -p $MODISCO_COUNTS_DIR
# mkdir -p $MODISCO_PROFILE_DIR

# modisco motifs\
#     --max_seqlets 1000000 \
#     --h5py $SHAP_DIR/profile_mean_shap_scores.h5 \
#     -o $MODISCO_PROFILE_DIR/profile_modisco_results.h5
# modisco report \
#     -i $MODISCO_PROFILE_DIR/profile_modisco_results.h5 \
#     -o $MODISCO_PROFILE_DIR/ \
#     -m ../data/motifs.meme.txt

modisco motifs\
    --max_seqlets 1000000 \
    --h5py $SHAP_DIR/counts_mean_shap_scores.h5 \
    -o $MODISCO_COUNTS_DIR/counts_modisco_results.h5
modisco report \
    -i $MODISCO_COUNTS_DIR/counts_modisco_results.h5 \
    -o $MODISCO_COUNTS_DIR/ \
    -m ../data/motifs.meme.txt