Ayush Shrivastava commited on
Commit
e90bc79
1 Parent(s): e6fee14

Changes to app file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,11 +17,11 @@ def model_MLP(X_train,y_train,X_test,layers, nodes, activation, solver, rate, it
17
  model = Sequential()
18
 
19
  # Adding first layers.
20
- model.add(Dense(nodes, activation=activation, kernel_initializer='he_uniform', input_dim=X_train.shape[1]))
21
 
22
  # Adding remaining hidden layers.
23
  for i in range(layers-1):
24
- model.add(Dense(nodes, activation=activation, kernel_initializer='he_uniform'))
25
 
26
  # Adding output layer.
27
  model.add(Dense(1, activation='linear'))
 
17
  model = Sequential()
18
 
19
  # Adding first layers.
20
+ model.add(Dense(nodes, activation=activation, input_dim=X_train.shape[1]))
21
 
22
  # Adding remaining hidden layers.
23
  for i in range(layers-1):
24
+ model.add(Dense(nodes, activation=activation))
25
 
26
  # Adding output layer.
27
  model.add(Dense(1, activation='linear'))