| name: "LogisticRegressionNet" | |
| layer { | |
| name: "data" | |
| type: "HDF5Data" | |
| top: "data" | |
| top: "label" | |
| include { | |
| phase: TRAIN | |
| } | |
| hdf5_data_param { | |
| source: "examples/hdf5_classification/data/train.txt" | |
| batch_size: 10 | |
| } | |
| } | |
| layer { | |
| name: "data" | |
| type: "HDF5Data" | |
| top: "data" | |
| top: "label" | |
| include { | |
| phase: TEST | |
| } | |
| hdf5_data_param { | |
| source: "examples/hdf5_classification/data/test.txt" | |
| batch_size: 10 | |
| } | |
| } | |
| layer { | |
| name: "fc1" | |
| type: "InnerProduct" | |
| bottom: "data" | |
| top: "fc1" | |
| param { | |
| lr_mult: 1 | |
| decay_mult: 1 | |
| } | |
| param { | |
| lr_mult: 2 | |
| decay_mult: 0 | |
| } | |
| inner_product_param { | |
| num_output: 40 | |
| weight_filler { | |
| type: "xavier" | |
| } | |
| bias_filler { | |
| type: "constant" | |
| value: 0 | |
| } | |
| } | |
| } | |
| layer { | |
| name: "relu1" | |
| type: "ReLU" | |
| bottom: "fc1" | |
| top: "fc1" | |
| } | |
| layer { | |
| name: "fc2" | |
| type: "InnerProduct" | |
| bottom: "fc1" | |
| top: "fc2" | |
| param { | |
| lr_mult: 1 | |
| decay_mult: 1 | |
| } | |
| param { | |
| lr_mult: 2 | |
| decay_mult: 0 | |
| } | |
| inner_product_param { | |
| num_output: 2 | |
| weight_filler { | |
| type: "xavier" | |
| } | |
| bias_filler { | |
| type: "constant" | |
| value: 0 | |
| } | |
| } | |
| } | |
| layer { | |
| name: "loss" | |
| type: "SoftmaxWithLoss" | |
| bottom: "fc2" | |
| bottom: "label" | |
| top: "loss" | |
| } | |
| layer { | |
| name: "accuracy" | |
| type: "Accuracy" | |
| bottom: "fc2" | |
| bottom: "label" | |
| top: "accuracy" | |
| include { | |
| phase: TEST | |
| } | |
| } | |