package org.deeplearning4j.nn.modelimport.keras.config;

import lombok.Data;

@Data
public class KerasModelConfiguration {

    /* Model meta information fields */
    private final String fieldClassName = "class_name";
    private final String fieldClassNameSequential = "Sequential";
    private final String fieldClassNameModel = "Model";
    private final String fieldKerasVersion = "keras_version";

    /* Model configuration field. */
    private final String modelFieldConfig = "config";
    private final String modelFieldLayers = "layers";
    private final String modelFieldInputLayers = "input_layers";
    private final String modelFieldOutputLayers = "output_layers";

    /* Training configuration field. */
    private final String trainingLoss = "loss";
    private final String trainingWeightsRoot = "model_weights";
    private final String trainingModelConfigAttribute = "model_config";
    private final String trainingTrainingConfigAttribute = "training_config";

}

@@ -1,81 +1,850 @@
-package org.deeplearning4j.nn.modelimport.keras.config;%0A%0Aimport lombok.Data;%0A
+/*-%0A *%0A *  * Copyright 2017 Skymind,Inc.%0A *  *%0A *  *    Licensed under the Apache License, Version 2.0 (the %22License%22);%0A *  *    you may not use this file except in compliance with the License.%0A *  *    You may obtain a copy of the License at%0A *  *%0A *  *        http://www.apache.org/licenses/LICENSE-2.0%0A *  *%0A *  *    Unless required by applicable law or agreed to in writing, software%0A *  *    distributed under the License is distributed on an %22AS IS%22 BASIS,%0A *  *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A *  *    See the License for the specific language governing permissions and%0A *  *    limitations under the License.%0A *%0A */%0Apackage org.deeplearning4j.nn.modelimport.keras.config;%0A%0Aimport lombok.Data;%0A%0A%0A/**%0A * Basic properties and field names of serialised Keras models.%0A *%0A * @author Max Pumperla%0A */
 %0A@Da

add copyright and docstring