/** * setup the local DBN instance based on conf params * */ @Override public void setup(Configuration conf) { NeuralNetConfiguration conf2 = NeuralNetConfiguration.fromJson(conf.get(NEURAL_NET_CONF)); neuralNetwork = conf2.getLayerFactory().create(conf2); }
/** * JSON model configuration passed in * If you are entering a MultiLayerConfiguration JSON, * your file name MUST contain '_multi'. * Otherwise, it will be processed as a regular * NeuralNetConfiguration * * Takes in JSON file path * Checks file path for indication of MultiLayer * Reads JSON file to string * Creates neural net configuration from string config * */ @Override public <E> E value(String value) throws Exception { Boolean isMultiLayer = value.contains("_multi"); String json = FileUtils.readFileToString(new File(value)); if (isMultiLayer) { return (E) MultiLayerConfiguration.fromJson(json); } else { return (E) NeuralNetConfiguration.fromJson(json); } }
@Override public void setup(Configuration conf) { NeuralNetConfiguration conf2 = NeuralNetConfiguration.fromJson(conf.get(NEURAL_NET_CONF)); this.neuralNetwork = conf2.getLayerFactory().create(conf2); } }
NeuralNetConfiguration conf = NeuralNetConfiguration.fromJson(FileUtils.readFileToString(new File(modelPath))); LayerFactory factory = LayerFactories.getFactory(conf); int numParams = LayerFactories.getFactory(conf).initializer().numParams(conf,true);