Tensorboard не показывает все данные тренировки - PullRequest
Новая
       4

Tensorboard не показывает все данные тренировки

1 голос
/ 10 февраля 2020

Здравствуйте, я довольно новичок в нейронной сети. Я пытаюсь построить MAE моих данных обучения и проверки через тензорную доску. Тем не менее, тензорная доска показывает только небольшую часть тренировочных данных. Это показывает все данные проверки, хотя. Я также строю историю, используя matplotlib, и все правильно. Ниже мой код и график. В чем проблема и как ее исправить? Заранее спасибо. 

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
from tensorflow import keras
from datetime import datetime


#  ---------train the neural network-----------
model = keras.Sequential([  # Sequential training function
    keras.layers.Dense(30, activation="relu", kernel_initializer='random_uniform',
                       bias_initializer='zeros', input_shape=(2,), name="input"),  # input layer
    keras.layers.Dense(30, activation="relu", name="hidden_1"),  # the first hidden layer with 30 neurons
    keras.layers.Dense(30, activation="relu", name="hidden_2"),  # the second hidden layer with 30 neurons
    keras.layers.Dense(1, name="output")  # the output layer with 1 neuron since there is only one output
])
epochs = 500
optimizer = keras.optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=True)  # define the optimizer
tb = keras.callbacks.TensorBoard(log_dir=r"C:\Users\Beichao\Desktop\BEICHAO\Neural Network\python code\regression"
                                 .format(datetime.now()))

model.compile(loss="mean_squared_error", optimizer=optimizer, metrics=['mae'])  # define some parameters
history = model.fit(x_train, y_train, batch_size=32, validation_data=[x_validation, y_validation]
                    , epochs=epochs, verbose=0, callbacks=[tb])  # train the NN
error = model.evaluate(x_validation, y_validation)
print(f"mean absolute error is {error}")

фигура тенорборда

фигура matplotlib

...