График полного к одной функции потери - PullRequest
Новая
       2

График полного к одной функции потери

0 голосов
/ 17 января 2020

Когда я строю функцию потерь своего кода, я получаю хороший график потерь. Если я хочу построить 1-hist.history['loss'], как я могу это сделать? Часть моего кода: 

model = Sequential([
    Dense(32, activation='relu', input_shape=(2,)),
    Dense(32, activation='relu'),
    Dense(1, activation='relu'), ])

model.compile(optimizer='sgd',
             loss='mean_squared_error')

hist = model.fit(X_train, Y_train,
                 batch_size=32, epochs=100,
                 validation_data=(X_val, Y_val))

A = model.evaluate(X_test, Y_test)[1]
print(A)  

plt.plot(hist.history['loss'])
plt.plot(hist.history['val_loss'])
plt.title('Model loss')
plt.ylabel('Loss')
plt.xlabel('Epoch')
plt.legend(['Train', 'Val'], loc='upper right')
plt.savefig("loss.png")  # Save the plot of loss function
plt.clf()

В этой части кода: plt.plot(hist.history['loss']) Я хочу изменить его на число, которое завершается до 1.

Есть ли способ сделать это

1 Ответ

1 голос
/ 17 января 2020

Что-то попробовать, если hist - это pandas фрейм данных: 

complete_to_one = 1 - hist.history['loss'])
plt.plot(complete_to_one)
...