Python SparkMLlibModel.save示例

示例#1

def test_serialization():
    spark_model = SparkMLlibModel(model,
                                  frequency='epoch',
                                  mode='synchronous',
                                  num_workers=2)
    spark_model.save("test.h5")
    load_spark_model("test.h5")

示例#2

文件： test_mllib_model.py 项目： rishabh706/elephas

def test_serialization(classification_model):
    rms = RMSprop()
    classification_model.compile(rms, 'categorical_crossentropy', ['acc'])
    spark_model = SparkMLlibModel(
        classification_model, frequency='epoch', mode='synchronous', num_workers=2)
    spark_model.save("test.h5")
    loaded_model = load_spark_model("test.h5")
    assert loaded_model.master_network.to_yaml()