trohith89 commited on
Commit
25d32d1
·
verified ·
1 Parent(s): 66dec37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -275,7 +275,7 @@ def create_ann_model(input_dim, hidden_layers, neurons_per_layer):
275
  model.add(Dense(1, activation='sigmoid'))
276
 
277
  # Compile the model with explicit learning rate
278
- optimizer = SGD(learning_rate=learning_rate, batch_size = batch_size)
279
  model.compile(
280
  optimizer=optimizer,
281
  loss=BinaryCrossentropy(),
 
275
  model.add(Dense(1, activation='sigmoid'))
276
 
277
  # Compile the model with explicit learning rate
278
+ optimizer = SGD(learning_rate=learning_rate)
279
  model.compile(
280
  optimizer=optimizer,
281
  loss=BinaryCrossentropy(),