#!/bin/bash
set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "${SCRIPT_DIR}/.." && pwd)"
cd "${PROJECT_ROOT}"

mkdir -p runs

# -------------------------
# Late AMD
# -------------------------
python survival/train_cox.py \
  --json data/AREDS1_all_survival_small_Status_late_amd_20190601.json \
  --endpoint late_amd \
  --feature-set grading \
  --output-dir runs/stage1_late_amd_grading

python survival/train_cox.py \
  --json data/AREDS1_all_survival_small_Status_late_amd_20190601.json \
  --endpoint late_amd \
  --feature-set grading_clinical \
  --output-dir runs/stage1_late_amd_grading_clinical

# -------------------------
# Any GA
# -------------------------
python survival/train_cox.py \
  --json data/AREDS1_all_survival_small_Status_anyga_20190601.json \
  --endpoint anyga \
  --feature-set grading \
  --output-dir runs/stage1_anyga_grading

python survival/train_cox.py \
  --json data/AREDS1_all_survival_small_Status_anyga_20190601.json \
  --endpoint anyga \
  --feature-set grading_clinical \
  --output-dir runs/stage1_anyga_grading_clinical

# -------------------------
# NV
# -------------------------
python survival/train_cox.py \
  --json data/AREDS1_all_survival_small_Status_nv_20190601.json \
  --endpoint nv \
  --feature-set grading \
  --output-dir runs/stage1_nv_grading

python survival/train_cox.py \
  --json data/AREDS1_all_survival_small_Status_nv_20190601.json \
  --endpoint nv \
  --feature-set grading_clinical \
  --output-dir runs/stage1_nv_grading_clinical