把tensorboard的输出文件改成自己想要的,
(1)记得有斜杆要转义,也就是用双斜杠,好像是tensorflow2.0的bug。
(2)要用绝对路径,不能用相对路径,原因不记得了
from keras.callbacks import TensorBoard
import datetime
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense,Dropout,Flatten
#第二种数据集加载方式,用数据集文件的绝对路径
path = 'E:/python_work1/learn/data/keras_datasets/mnist.npz'
(x_train, y_train),(x_test, y_test) = mnist.load_data(path)
x_train, x_test = x_train / 255.0, x_test / 255.0
model = Sequential()
model.add(Flatten(input_shape=(28,28)))
model.add(Dense(512,activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(10, activation='softmax'))
model.summary()
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
log_dir="E:\\python_work1\\learn\\tensorboard\\test\\" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = TensorBoard(log_dir=log_dir, histogram_freq=1)
model.fit(x=x_train,
y=y_train,
epochs=5,
validation_data=(x_test, y_test),
callbacks=[tensorboard_callback])