--- a
+++ b/rf/args.py
@@ -0,0 +1,43 @@
+#!/usr/bin/env python
+
+from argparse import Namespace
+from pathlib import Path
+
+path = Path('../data')
+workdir=path/'workdir/rf'
+
+args = Namespace(
+  workdir=workdir,
+  dataset_csv=path/'proc_dataset.csv',
+  cols=['hadm_id', 'imminent_adm_label', 'prolonged_stay_label', 'processed_note', 'charttime', 'intime', 'chartinterval'],
+  imminent_adm_cols=['hadm_id', 'processed_note', 'imminent_adm_label'],
+  prolonged_stay_cols=['hadm_id', 'processed_note', 'prolonged_stay_label'],
+  dates=['charttime', 'intime'],
+  modeldir=workdir/'models',
+  min_freq=3,
+  ia_thresh=0.27,
+  ps_thresh=0.37,
+  start_seed=127,
+  )
+
+ia_params = {
+  "class_weight": 'balanced',
+  "criterion": "entropy",
+  "max_features": 0.12,
+  "min_samples_leaf": 3,
+  "min_samples_split": 7,
+  "n_estimators": 108,
+  "oob_score": True,
+  "n_jobs": 32,
+}
+
+ps_params = {
+  "class_weight": 'balanced',
+  "criterion": "entropy",
+  "max_features": 0.114,
+  "min_samples_leaf": 5,
+  "min_samples_split": 3,
+  "n_estimators": 116,
+  "oob_score": True,
+  "n_jobs": 32,
+}