Skip to content

Commit fe45f11

Browse files
committed
[tmva][sofie] Remove batch_size keyword arument from Dense layers
This keyword argument is not relevant for defining the model and will cause errors in newer Keras versions.
1 parent ff486c4 commit fe45f11

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tmva/sofie/test/generateKerasModels.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def generateFunctionalModel():
2828

2929
def generateSequentialModel():
3030
model=Sequential()
31-
model.add(Dense(8,batch_size=4))
31+
model.add(Dense(8))
3232
model.add(ReLU())
3333
model.add(Dense(6))
3434
model.add(Activation('sigmoid'))
@@ -43,7 +43,7 @@ def generateSequentialModel():
4343

4444
def generateBatchNormModel():
4545
model=Sequential()
46-
model.add(Dense(4,batch_size=2))
46+
model.add(Dense(4))
4747
model.add(BatchNormalization())
4848
model.add(Dense(2))
4949

@@ -180,4 +180,4 @@ def generateCustomModel():
180180
generateBinaryOpModel()
181181
generateActivationModel()
182182
generateSwishModel()
183-
generateCustomModel()
183+
generateCustomModel()

0 commit comments

Comments
 (0)