我现在可以把我的Keras数据(模型,kappa,损耗,历元,...)输入到MLflow中了,我原来的模型是H5,但是MLflow中artifacts/model/data/model/saved_model.pb下保存的模型是PB,可以在MLflow中保存.h5吗?
x0fgdtte1#
您可以保存和记录模型及其参数和指标。两者都保留Keras HDF5格式,如MLflow Keras文档中所述。
1条答案
按热度按时间x0fgdtte1#
您可以保存和记录模型及其参数和指标。两者都保留Keras HDF5格式,如MLflow Keras文档中所述。