7567cc646d258e40dde9790a28a9b264ccd494fb,mimic3models/split_train_val.py,,,#,5

Before Change

import shutil
import argparse
import random
random.seed(47297)

parser = argparse.ArgumentParser(description="Split train data into train and validation sets.")
parser.add_argument("task", type=str, help="Possible values are: decompensation, "\
                            "in-hospital-mortality, length-of-stay, phenotyping, multitask.")

After Change

                     "phenotyping", "multitask"]

val_patients = set()
with open("mimic3models/valset.csv", "r") as valset_file:
    for line in valset_file:
        x, y = line.split(",")
        if int(y) == 1:
            val_patients.add(x)

has_header = False
if args.task in ["phenotyping", "multitask"]:
    has_header = True

Instances