train.sh

#!/bin/bash

python train.py \
--sequence_embedding sequence_embedding \
--input_dir data \
--num_classes 2 \
--device 'cuda:0' \
--num_train_patients 200 \
--num_sequences 200