概述
模型导入概述
DL4J: Keras模型导入
入门:在60秒内导入一个Keras模型
from keras.models import Sequential
from keras.layers import Dense
model = Sequential()
model.add(Dense(units=64, activation='relu', input_dim=100))
model.add(Dense(units=10, activation='softmax'))
model.compile(loss='categorical_crossentropy',optimizer='sgd', metrics=['accuracy'])
model.save('simple_mlp.h5')String simpleMlp = new ClassPathResource("simple_mlp.h5").getFile().getPath();
MultiLayerNetwork model = KerasModelImport.importKerasSequentialModelAndWeights(simpleMlp);项目设置
后端
流行的模型与应用
故障排除与支持
为什么需要Keras模型导入?
Last updated
Was this helpful?