Keras実装練習:線形関数フィット(1)

◎環境：Python 3.6.7、TensorFlow 1.13.1

概要

Kerasの実装練習として、線形関数の回帰をやってみる。まずはざっと実装し、実行してみる。

実装のポイント

入力データ

# 入力データ(訓練用)
x_train = np.linspace(-10,10,1000)
a = 2
b = -3
y_train = a*x_train + b

# 入力データ(テスト用)
x_test = np.linspace(-10,10,101)
y_test = a*x_test + b

モデル

model = Sequential()
model.add(Dense(3,input_shape=(1,)))
model.add(Dense(1))

model.compile(loss='mean_squared_error',
             optimizer=keras.optimizers.RMSprop(),
             metrics=['accuracy'])

batch_size = 128
epochs = 100

history = model.fit(x_train,y_train,
                   batch_size=batch_size,
                   epochs=epochs,
                   verbose=1,
                   validation_data=(x_test,y_test))

学習結果

f:id:storikai:20190522000749p:plain