как визуализировать модель DNN или CNN со значениями входного и выходного слоев?
Я применил визуализацию к модели DNN, но изображение просто содержит плотный слой без значения входных и выходных слоев! Приведенный ниже код объясняет процесс визуализации без каких-либо ошибок, я попытался показать значения входных и выходных слоев на изображении.
import pandas as pd
.
.
tf.keras.utils.plot_model
.
.
def create_model():
model = Sequential()
model.add(Input(n_features))
model.add(BatchNormalization())
model.add(Dense(51, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(68, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(85, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(85, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(68, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(51, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(n_outputs, activation='sigmoid'))
model.compile(loss='binary_crossentropy',
optimizer='Adam',
metrics=['accuracy'])
tf.keras.utils.plot_model(model, to_file='model_combined.png')
#model.summary()
return model
#I have tried to use
#from keras.utils.vis_utils import plot_model
# but i found this error : TypeError: 'InputLayer' object is not iterable
# so i use the above library to implement visualization without any error.
Обратите внимание, что я загрузил все эти библиотеки: Graphviz, pydot, pydotplus, python-graphviz.
1 ответ
Вы пробовали приложение Netron ? Его можно использовать для визуализации модели DNN или CNN со значениями входного и выходного слоев.
Здесь вы можете найти все подробности