4034844b505f20da652a417e605d391eae8d6c0a,scripts/split_train_and_test.py,,,#,5

Before Change

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


parser = argparse.ArgumentParser(description="Split data into train and test sets.")
parser.add_argument("subjects_root_path", type=str, help="Directory containing subject sub-directories.")

After Change

args, _ = parser.parse_known_args()

testset = set()
with open("resources/testset.csv", "r") as test_set_file:
    for line in test_set_file:
        x, y = line.split(",")
        if int(y) == 1:
            testset.add(x)

def move_to_partition(patients, partition):
    if not os.path.exists(os.path.join(args.subjects_root_path, partition)):
        os.mkdir(os.path.join(args.subjects_root_path, partition))
    for patient in patients:

Instances