7b6fd231d17cd1fd88c1fc163a33f1049d63e7b0,examples/convolutional_model.py,,,#,3

Before Change

print("Creating training/testing data...")
validation_split = 0.15
X_train, X_test, y_train, y_test = train_test_split(images, labels,
                                                    test_size=validation_split, random_state=42, stratify=labels)
train_gen = DataGenerator().fit(X_train, y_train)
test_gen = DataGenerator().fit(X_test, y_test)

After Change

dataset = data_loader.load_data()

if verbose:
    dataset.print_data_details()

print("Preparing training/testing data...")
train_images, train_labels = dataset.get_training_data()
train_gen = DataGenerator().fit(train_images, train_labels)

Instances