From 89a49b9e93c5dd1116fd4abd021b6ae89a049c06 Mon Sep 17 00:00:00 2001 From: Kartik Chugh Date: Fri, 6 Nov 2020 18:02:30 -0500 Subject: [PATCH] Clean up default (non-pretrained) CNN model code --- libra/query/feedforward_nn.py | 16 +--------------- 1 file changed, 1 insertion(+), 15 deletions(-) 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,