Skip to content

Commit d02803f

Browse files
committed
fixed pool size
1 parent d1488fc commit d02803f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

examples/deep_bayesian_active_learning.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def create_keras_model():
1313
model = Sequential()
1414
model.add(Conv2D(32, (3, 3), activation='relu'))
1515
model.add(Conv2D(32, (3, 3), activation='relu'))
16-
model.add(MaxPooling2D(pool_size=(2, 2)))
16+
model.add(MaxPooling2D(pool_size=(5, 5)))
1717
model.add(Dropout(0.25))
1818
model.add(Flatten())
1919
model.add(Dense(128, activation='relu'))
@@ -96,5 +96,6 @@ def uniform(learner, X, n_instances=1):
9696
print('Accuracy after query {n}: {acc:0.4f}'.format(n=index + 1, acc=model_accuracy))
9797
perf_hist = [model_accuracy]
9898

99-
np.save('/home/damien/Results/keras_modal_entropy.npy', perf_hist)
100-
print("saving to /home/damien/Results/keras_modal_entropy.npy")
99+
save_path = "/home/damien/Results/keras_modal_riashat_entropy.npy"
100+
np.save(save_path, perf_hist)
101+
print(save_path)

0 commit comments

Comments
 (0)