KERNEL_INIT = 'glorot_normal' CONV_DROPOUT = 0.1 POOLING = 'max' REGPEN=1e-4 seq.add(Conv3D(12, (2, 2, 2), #strides=(2,2), activation='relu', input_shape=input_shape, kernel_initializer=KERNEL_INIT, kernel_regularizer=l2(REGPEN))) seq.add(BatchNormalization()) seq.add(Conv3D(8, (2, 2, 2), activation='relu', kernel_regularizer=l2(REGPEN), kernel_initializer=KERNEL_INIT, bias_initializer=TruncatedNormal())) if POOLING == 'avg': seq.add(AveragePooling3D()) elif POOLING == 'max': seq.add(MaxPooling3D()) else: alg_notes += ' no pooling' seq.add(Conv3D(64,(2,2,2),activation='relu', kernel_initializer=KERNEL_INIT, kernel_regularizer=l2(REGPEN), bias_initializer=TruncatedNormal())) seq.add(Flatten()) seq.add(Dense(dense_size, activation='relu', kernel_initializer=KERNEL_INIT)) seq.add(Dense(128, activation='relu', kernel_initializer=KERNEL_INIT))