5d353701dd56a1fc8abc15e4082e33b7bed2a241,mimic3models/split_train_val.py,,,#,5

Before Change

val_lines = [x for x in lines if x[:x.find("_")] in val_patients]
assert len(train_lines) + len(val_lines) == len(lines)

if not os.path.exists("data/%s/" % args.task):
    os.makedirs("data/%s/" % args.task)

with open("data/%s/train_listfile.csv" % args.task, "w") as train_listfile:
    if has_header:
        train_listfile.write(header)
    for line in train_lines:

After Change

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