Manul, da
commited on
Commit
•
8419ac2
1
Parent(s):
1042d9f
Update net.py
Browse files
net.py
CHANGED
@@ -35,7 +35,7 @@ model.add(Dense(200, activation="selu"))
|
|
35 |
model.add(Dense(128, activation="softplus"))
|
36 |
model.add(Dense(1, activation="softplus"))
|
37 |
|
38 |
-
model.compile(optimizer=Adam(learning_rate=0.
|
39 |
|
40 |
model.fit(X, y, epochs=2, batch_size=4, workers=4, use_multiprocessing=True)
|
41 |
|
|
|
35 |
model.add(Dense(128, activation="softplus"))
|
36 |
model.add(Dense(1, activation="softplus"))
|
37 |
|
38 |
+
model.compile(optimizer=Adam(learning_rate=0.001), loss="mse", metrics=["accuracy",])
|
39 |
|
40 |
model.fit(X, y, epochs=2, batch_size=4, workers=4, use_multiprocessing=True)
|
41 |
|