| 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: 2 |
| weight_filler { |
| type: "xavier" |
| } |
| bias_filler { |
| type: "constant" |
| value: 0 |
| } |
| } |
| } |
| layer { |
| name: "loss" |
| type: "SoftmaxWithLoss" |
| bottom: "fc1" |
| bottom: "label" |
| top: "loss" |
| } |
| layer { |
| name: "accuracy" |
| type: "Accuracy" |
| bottom: "fc1" |
| bottom: "label" |
| top: "accuracy" |
| include { |
| phase: TEST |
| } |
| } |
|
|