package skymind.dsx;

/**
 * Created by tomhanlon on 12/29/17.
 */
import org.apache.log4j.BasicConfigurator;
import org.deeplearning4j.nn.modelimport.keras.KerasModelImport;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;


public class Keras2ModelImport {
    public static void main(String[] args) throws Exception{
        BasicConfigurator.configure();



        String kerasModelfromKerasExport = "/tmp/my_model.h5";


        MultiLayerNetwork model = KerasModelImport.importKerasSequentialModelAndWeights(kerasModelfromKerasExport);


        System.out.println("Model Configuration");
        String modelconfig = model.conf().toJson();
        System.out.println(modelconfig);



    }
}