#!/usr/bin/env python

from argparse import Namespace
from pathlib import Path

path = Path('../data')

args = Namespace(
  workdir=path/'workdir',
  figdir=path/'results/figures',
  resultdir=path/'results',
  dataset_csv=path/'proc_dataset.csv',
  imminent_adm_cols=['hadm_id', 'processed_note', 'imminent_adm_label'],
  prolonged_stay_cols=['hadm_id', 'processed_note', 'prolonged_stay_label'],
  cols=['hadm_id', 'imminent_adm_label', 'prolonged_stay_label', 'processed_note', 'charttime', 'intime', 'chartinterval'],
  dates=['charttime', 'intime'],
  ia_thresh={
    'lr': 0.45,
    'rf': 0.27,
    'gbm': 0.435,
    'mlp': 0.2,
    },
  ps_thresh={
    'lr': 0.39,
    'rf': 0.36,
    'gbm': 0.324,
    'mlp': 0.27,
    },
)