Merge pull request #45 from EmbraceLife/master

update Dense with units
MorvanZhou · May 13, 2017 · 7269d08 · 7269d08
2 parents 2072e89 + 309b821
commit 7269d08
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/kerasTUT/4-regressor_example.py b/kerasTUT/4-regressor_example.py
@@ -29,7 +29,8 @@
 
 # build a neural network from the 1st layer to the last layer
 model = Sequential()
-model.add(Dense(output_dim=1, input_dim=1))
+
+model.add(Dense(units=1, input_dim=1)) 
 
 # choose loss function and optimizing method
 model.compile(loss='mse', optimizer='sgd')
@@ -52,4 +53,4 @@
 Y_pred = model.predict(X_test)
 plt.scatter(X_test, Y_test)
 plt.plot(X_test, Y_pred)
-plt.show()
+plt.show()