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])

留言

這個網誌中的熱門文章

VMware Workstation 教學之三種網絡設定模式

SET命令詳解

Elastix 呼入路由設定 (Inbound Routes)