#!/bin/bash
#SBATCH --job-name=probe10v2
#SBATCH --partition=nova
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=4
#SBATCH --gres=gpu:v100:1
#SBATCH --mem=32G
#SBATCH --time=00:30:00
#SBATCH --output=bert_v6_contrastive/additional_probes/probe10v2_%j.out
#SBATCH --error=bert_v6_contrastive/additional_probes/probe10v2_%j.err

echo "========================================"
echo "Probe 10 v2: Tissue-Specific Glycan Presence"
echo "  Data: 243 glycans (best WURCS + recovered)"
echo "  Well-powered: 16 tissues (>=30 positives)"
echo "Job ID: $SLURM_JOB_ID"
echo "Node:   $HOSTNAME"
echo "Start:  $(date)"
echo "========================================"

source activate blade 2>/dev/null || conda activate blade

cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training

SCRIPT=bert_v6_contrastive/additional_probes/probe_10_tissue_presence.py
DATA_V2=bert_v6_contrastive/additional_probes/probe_10_tissue_presence_data_v2.csv

python3 -c "import torch; print(f'PyTorch: {torch.__version__} CUDA: {torch.cuda.is_available()}')"
python3 -c "import torch; print(f'{torch.cuda.get_device_name(0)}, {torch.cuda.get_device_properties(0).total_memory//1024**2} MiB')" 2>/dev/null

echo ""
ls -lh $DATA_V2
wc -l $DATA_V2
echo ""

echo "========== V5-A: v2 data, all 23 tissues =========="
python3 $SCRIPT --model v5 --data-version v2
echo ""

echo "========== V5-A: v2 data, well-powered (16 tissues) =========="
python3 $SCRIPT --model v5 --data-version v2 --well-powered
echo ""

echo "========== V6: v2 data, all 23 tissues =========="
python3 $SCRIPT --model v6 --data-version v2
echo ""

echo "========== V6: v2 data, well-powered (16 tissues) =========="
python3 $SCRIPT --model v6 --data-version v2 --well-powered
echo ""

echo "Completed: $(date)"
echo "Output dirs:"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v5_v2/ 2>/dev/null
echo "---"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v5_v2_wp/ 2>/dev/null
echo "---"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v6_v2/ 2>/dev/null
echo "---"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v6_v2_wp/ 2>/dev/null