Merge pull request #372 from KartikChugh/patch-1

Looks pretty straight forward. Clean up default (non-pretrained) CNN model code
Palashio · Nov 9, 2020 · 0c055fa · 0c055fa
2 parents 7c80620 + 89a49b9
commit 0c055fa
Showing 1 changed file with 1 addition and 15 deletions.
diff --git 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,