Как вывести значение потерь в tenorflow 2 (python)? - PullRequest
Новая
       85

Как вывести значение потерь в tenorflow 2 (python)?

0 голосов
/ 29 апреля 2020

Я тренирую двухслойную нейронную сеть. Я хочу запустить модель с разными скоростями обучения, чтобы найти оптимальное значение. Можно ли напечатать значение потерь для каждой скорости обучения?

Это код, который я использую: 

(X_train_orig,y_train_orig),(X_test_orig,y_test_orig) = fashion_mnist.load_data()
    X_train = X_train_orig / 255
    X_test = X_test_orig / 255
    X_train = X_train.reshape(X_train_orig.shape[0],-1)
    X_test = X_test.reshape(X_test_orig.shape[0],-1)
    model = tf.keras.models.Sequential()
    model.add(tf.keras.layers.Dense(units=128,activation='relu',input_shape=(784, )))
    model.add(tf.keras.layers.Dropout(0.2))
    model.add(tf.keras.layers.Dense(units=10,activation='softmax'))
    np.random.seed(1)
    r = -4 * np.random.rand(10)
    r = list(r)
    learning_rate = np.power(10,r)
    for i in range(10):
        optimizer = tf.keras.optimizers.Adam(lr=learning_rate[i])
        model.compile(optimizer=optimizer,loss='sparse_categorical_crossentropy',metrics=['sparse_categorical_accuracy'])
        model.fit(X_train,y_train,epochs=100)
...