Python ElephasEstimator.save示例

示例#1

文件： test_ml_model.py 项目： blakermchale/parallel-stocks

def test_serialization_estimator():
    estimator = ElephasEstimator()
    estimator.set_keras_model_config(model.to_yaml())
    estimator.set_loss("categorical_crossentropy")

    estimator.save("test.h5")
    load_ml_estimator("test.h5")

示例#2

def test_serialization_estimator(classification_model):
    estimator = ElephasEstimator()
    estimator.set_keras_model_config(classification_model.to_yaml())
    estimator.set_loss("categorical_crossentropy")

    estimator.save("test.h5")
    loaded_model = load_ml_estimator("test.h5")
    assert loaded_model.get_model().to_yaml() == classification_model.to_yaml()