Có thể trực quan hóa các đoạn nhúng keras trong tensorboard không?

Question

keras có khả năng xuất một số dữ liệu đào tạo của nó ở định dạng comaptible tensorboard bằng cách sử dụng keras.callbacks.TensorBoard

Tuy nhiên, nó không hỗ trợ embedding visualisation trong bảng dẫn.

Có cách nào khác không?

Answer 1

Tìm thấy một giải pháp:

import os 

import keras 
import tensorflow 

ROOT_DIR = '/tmp/tfboard' 

os.makedirs(ROOT_DIR, exist_ok=True) 


OUTPUT_MODEL_FILE_NAME = os.path.join(ROOT_DIR,'tf.ckpt') 

# get the keras model 
model = get_model() 
# get the tensor name from the embedding layer 
tensor_name = next(filter(lambda x: x.name == 'embedding', model.layers)).W.name 

# the vocabulary 
metadata_file_name = os.path.join(ROOT_DIR,tensor_name) 

embedding_df = get_embedding() 
embedding_df.to_csv(metadata_file_name, header=False, columns=[]) 

saver = tensorflow.train.Saver() 
saver.save(keras.backend.get_session(), OUTPUT_MODEL_FILE_NAME) 

summary_writer = tensorflow.train.SummaryWriter(ROOT_DIR) 

config = tensorflow.contrib.tensorboard.plugins.projector.ProjectorConfig() 
embedding = config.embeddings.add() 
embedding.tensor_name = tensor_name 
embedding.metadata_path = metadata_file_name 
tensorflow.contrib.tensorboard.plugins.projector.visualize_embeddings(summary_writer, config) 

Answer 2

Có yêu cầu kéo với chức năng này - https://github.com/fchollet/keras/pull/5247 callback được mở rộng để tạo ra hình dung cho các lớp nhúng cụ thể.