Merge pull request MorvanZhou#45 from EmbraceLife/master

MorvanZhou · web-flow · commit 7269d08a5452 · 2017-05-13T11:53:19.000+10:00
update Dense with units
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()
