Тензорная доска, как указать, какую модель использовать и какие цвета использовать? - PullRequest
0 голосов
/ 17 января 2019

Я пытаюсь запустить Tensorboard и использую проектор для просмотра встраивания слоев.
Но после того, как я запустил тензорную доску с помощью следующей команды, тренировка прекращается:

tensorboard --logdir logs

Это происходит потому, что модель, указанная в контрольной точке файла, является текущей записываемой моделью. Так как Tensorboard получает доступ к файлу модели через указанную контрольную точку.
Кроме того, цвет в проекторе не то, что я ожидал. Вопрос:

1) Как я могу указать, какую контрольную точку модели следует использовать в проектор? Я ожидаю что-то вроде этого: tensorboard --logdir logs --model_path moel0.ckpt

2) как указать цвет для получения данных визуализируется в проекторе, в metadata.tsv

Я использовал Keras с тензорной доской как:

model = build_model(X.shape[1])
tensorboard = TensorBoard(log_dir="logs/",batch_size=10,
                          embeddings_freq=1,
                          embeddings_layer_names=['layer1','layer2'],
                          embeddings_metadata='metadata.tsv',
                          embeddings_data=X_train)
model.fit(X_train, Y_train, epochs=100, batch_size=500, verbose=True,callbacks=[tensorboard],validation_data=(X_test,Y_test))

save_model(model)

Вывод, который я вижу в проекте, выглядит примерно так:
output.
Я хочу получить цвета как красный, зеленый и аква. Как или где я могу это указать?

Пожалуйста, дайте мне знать.

...