tensorflow2.0 基本流程實例
以下代碼是使用的 Google colab 運行,python3.
!pip install tensorflow==2.0.0-alpha0 import numpy as np from tensorflow import keras # training data x_train = np.array([2, 4, 6, 8, 10, 12, 14], dtype=float) y_train = np.array([5, 9, 13, 17, 21, 25, 29], dtype=float) # model and layers model = keras.Sequential([keras.layers.Dense(units=1, input_shape=[1])]) # Model config model.compile(optimizer=keras.optimizers.Adam(0.1), loss = 'mean_squared_error') # Mode Training model.fit(x_train, y_train, epochs=500, verbose=False) # Mode testing model.predict([5])
留言
張貼留言