Spaces:
Sleeping
Sleeping
Ayush Shrivastava
commited on
Commit
·
e90bc79
1
Parent(s):
e6fee14
Changes to app file
Browse files
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,
|
| 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'))
|
|
|
|
| 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'))
|