b673a26860c1598e7ef66a301743a2ff1d96b61f,examples/keras_integration.py,,create_keras_model,#,16

Before Change

    Should be passed to KerasClassifier in the Keras scikit-learn API.
    
    model = Sequential()
    model.add(Dense(512, activation="relu", input_shape=(784, )))
    model.add(Dropout(0.2))
    model.add(Dense(512, activation="relu"))
    model.add(Dropout(0.2))

After Change

    Should be passed to KerasClassifier in the Keras scikit-learn API.
    
    model = Sequential()
    model.add(Conv2D(32, kernel_size=(3, 3), activation="relu", input_shape=(28, 28, 1)))
    model.add(MaxPooling2D(pool_size=(2, 2)))
    model.add(Dropout(0.25))
    model.add(Flatten())
    model.add(Dense(128, activation="relu"))

Instances