2b4dfea5674a6b6b4d6556d7546a387f41f947fa,tools/split_csv_to_train_test.py,,main,#,35

Before Change

    tag = os.path.basename(opts.path_source_dir)
    path_source_csv = os.path.join(opts.path_source_dir, "results.csv")
    path_train_csv = os.path.join(opts.path_save_dir, "{:s}_train.csv".format(tag))
    path_test_csv = os.path.join(opts.path_save_dir, "{:s}_test.csv".format(tag))
    df_source = pd.read_csv(path_source_csv)
    print("read csv:", path_source_csv)
    df_cleaned = clean_df(df_source)

After Change

    path_train_csv = os.path.join(opts.path_save_dir, "{:s}_train.csv".format(opts.name))
    path_test_csv = os.path.join(opts.path_save_dir, "{:s}_test.csv".format(opts.name))
    if not os.path.exists(opts.path_save_dir):
        os.makedirs(opts.path_save_dir)
    
    df_source = pd.read_csv(opts.path_source_csv)
    print("read csv:", opts.path_source_csv)
    df_cleaned = clean_df(df_source)
    df_train, df_test = shuffle_split_df(df_cleaned, opts.train_split, no_shuffle=opts.no_shuffle)

Instances