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'))
|