public class ElephasModelImport extends Object
Constructor and Description |
---|
ElephasModelImport() |
Modifier and Type | Method and Description |
---|---|
static SparkComputationGraph |
importElephasModelAndWeights(org.apache.spark.api.java.JavaSparkContext sparkContext,
String modelHdf5Filename)
Load Elephas model stored using model.save(...) in case that the underlying Keras
model is a functional `Model` instance, which corresponds to a DL4J SparkComputationGraph.
|
static SparkDl4jMultiLayer |
importElephasSequentialModelAndWeights(org.apache.spark.api.java.JavaSparkContext sparkContext,
String modelHdf5Filename)
Load Elephas model stored using model.save(...) in case that the underlying Keras
model is a functional `Sequential` instance, which corresponds to a DL4J SparkDl4jMultiLayer.
|
public static SparkComputationGraph importElephasModelAndWeights(org.apache.spark.api.java.JavaSparkContext sparkContext, String modelHdf5Filename) throws IOException, UnsupportedKerasConfigurationException, InvalidKerasConfigurationException
sparkContext
- Java SparkContextmodelHdf5Filename
- Path to HDF5 archive storing Elephas ModelIOException
- IO exceptionInvalidKerasConfigurationException
- Invalid Keras configUnsupportedKerasConfigurationException
- Unsupported Keras configSparkComputationGraph
public static SparkDl4jMultiLayer importElephasSequentialModelAndWeights(org.apache.spark.api.java.JavaSparkContext sparkContext, String modelHdf5Filename) throws IOException, UnsupportedKerasConfigurationException, InvalidKerasConfigurationException
sparkContext
- Java SparkContextmodelHdf5Filename
- Path to HDF5 archive storing Elephas modelIOException
- IO exceptionInvalidKerasConfigurationException
- Invalid Keras configUnsupportedKerasConfigurationException
- Unsupported Keras configSparkDl4jMultiLayer
Copyright © 2020. All rights reserved.