diff --git a/libra/query/feedforward_nn.py b/libra/query/feedforward_nn.py index 12fc8ef2..6878f17b 100644 --- a/libra/query/feedforward_nn.py +++ b/libra/query/feedforward_nn.py @@ -710,21 +710,7 @@ def convolutional(instruction=None, else: model = Sequential() - # model.add( - # Conv2D( - # 64, - # kernel_size=3, - # activation="relu", - # input_shape=input_shape)) - # model.add(MaxPooling2D(pool_size=(2, 2))) - # model.add(Conv2D(64, kernel_size=3, activation="relu")) - # model.add(MaxPooling2D(pool_size=(2, 2))) - # model.add(Flatten()) - # model.add(Dense(num_classes, activation="softmax")) - # model.compile( - # optimizer="adam", - # loss=loss_func, - # metrics=['accuracy']) + model.add(Conv2D( filters=64, kernel_size=5,