Вот как я сегментировал данные
(X, y), (X_test, y_test) = mnist.load_data()
X_train, X_val, y_train, y_val = train_test_split(X,
y,
test_size=0.1,
random_state=1)
Приведенный ниже код отображает X_train и X_val, я бы хотел добавить X_test для графика
history = model.fit(
x= X_train,
y= y_train,
epochs = 10,
validation_data = (X_val, y_val),
batch_size = 128
)
from matplotlib import pyplot as plt
plt.plot(history.history['loss'])
plt.plot(history.history['val_loss'])
plt.title('model loss')
plt.ylabel('loss')
plt.xlabel('epoch')
plt.legend(['train', 'val'], loc='upper left')
plt.show()
Можно ли построить график test_loss
поверх этого, чтобы получить обзор всех трех на одном участке?
Нет history.history['test_loss']
.