機械学習のココロ（Vol.60掲載）
著者：石井一夫
今回は、機械学習を実施する際に一番問題となる過学習の問題を取り上げます。過学習というのは、機械 学習のモデルが、ある特定の状況に過剰適合してしまい、新たなサンプルに対してうまく予測ができなくなるという現象です。
シェルスクリプトマガジン Vol.60は以下のリンク先でご購入できます。
図5 L1 正則化を実施するコード
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
l1_model = keras.models.Sequential([ keras.layers.Dense(16, kernel_regularizer=keras.regularizers.l1(0.001), activation=tf.nn.relu, input_shape=(NUM_WORDS,)), keras.layers.Dense(16, kernel_regularizer=keras.regularizers.l1(0.001), activation=tf.nn.relu), keras.layers.Dense(1, activation=tf.nn.sigmoid) ]) l1_model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy', 'binary_crossentropy']) l1_model_history = l1_model.fit(train_data, train_labels, epochs=20, batch_size=512, validation_data=(test_data, test_labels), verbose=2) |
図6 過学習の評価結果を表示するコード
|
1 2 |
plot_history([('baseline', baseline_history), ('l1', l1_model_history)]) |