def gen_train_val()

in processing/create_dataset.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

def gen_train_val(df, save_dir):
    for index, row in df.iterrows():
        file_name = row["index"]
        label = row["label"]
        file_dir = save_dir / label
        if not os.path.exists(file_dir):
            os.makedirs(file_dir)
        
        with open(f"{file_dir}/{file_name}.txt", 'w') as f:
            f.write(row["text"])