
    train_b = load_files(
        os.path.join(datadir, train_str), shuffle=True, categories=classes
    )
    if len(train_test_names) &gt; 1:
        test_str = train_test_names[1]
        test_b = load_files(
            os.path.join(datadir, test_str), shuffle=False, categories=classes
        )
        x_train = train_b.data
        y_train = train_b.target
        x_test = test_b.data
        y_test = test_b.target
    else:
        x_train, x_test, y_train, y_test = train_test_split(
            tra