How to save final model using keras?

前端 未结 6 661
一整个雨季
一整个雨季 2020-11-28 07:01

I use KerasClassifier to train the classifier.

The code is below:

import numpy
from pandas import read_csv
from keras.models import Sequential
from k         


        
6条回答
  •  囚心锁ツ
    2020-11-28 08:00

    you can save the model and load in this way.

    from keras.models import Sequential, load_model
    from keras_contrib.losses import import crf_loss
    from keras_contrib.metrics import crf_viterbi_accuracy
    
    # To save model
    model.save('my_model_01.hdf5')
    
    # To load the model
    custom_objects={'CRF': CRF,'crf_loss':crf_loss,'crf_viterbi_accuracy':crf_viterbi_accuracy}
    
    # To load a persisted model that uses the CRF layer 
    model1 = load_model("/home/abc/my_model_01.hdf5", custom_objects = custom_objects)
    

提交回复
热议问题