update Dense

EmbraceLife · EmbraceLife · commit e87279189c73 · 2017-05-13T07:12:21.000+08:00
diff --git a/kerasTUT/4-regressor_example.py b/kerasTUT/4-regressor_example.py
@@ -29,7 +29,7 @@
 
 # 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 +52,4 @@
 Y_pred = model.predict(X_test)
 plt.scatter(X_test, Y_test)
 plt.plot(X_test, Y_pred)
-plt.show()
+plt.show()
