Keras 常用工具函数

【画出模型】

1 from keras.utils import plot_model
2 plot_model(model,'fileName.png')

 【Durex编码】

from keras import utils
utils.to_categorical(y_train, nubber_of_classes)

 y：表示类别的数字

 number_of_classes：表示类别总数。

 

 

 

 

 

【表格显示模型结构】

1 model.summary()

 

转载于:https://www.cnblogs.com/ailex/p/9674765.html