/
dae_l2.yaml
27 lines (27 loc) · 1.2 KB
/
dae_l2.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
!obj:pylearn2.train.Train {
"dataset": !obj:pylearn2.datasets.dense_design_matrix.DenseDesignMatrix &dataset {
"X" : !obj:protein_transformer.transformer {'pickle_path':'./dae_l1.pkl', 'X': !obj:pattern_generator.load_dataset {'type':'binary','all_dataset':False,'dataset':'439'},},
},
"model": !obj:pylearn2.models.autoencoder.DenoisingAutoencoder {
"nvis" : %(nvis)i,
"nhid" : %(nhid)i,
"irange" : 0.05,
"corruptor": !obj:pylearn2.corruption.BinomialCorruptor {
corruption_level: .3,
},
"act_enc": "sigmoid",
"act_dec": "sigmoid", # Linear activation on the decoder side.
},
"algorithm": !obj:pylearn2.training_algorithms.sgd.SGD {
"learning_rate" : 1e-3,
"batch_size" : %(batch_size)i,
"monitoring_batches" : %(monitoring_batches)i,
"monitoring_dataset" : *dataset,
"cost" : !obj:pylearn2.costs.autoencoder.MeanSquaredReconstructionError {},
"termination_criterion" : !obj:pylearn2.termination_criteria.EpochCounter {
"max_epochs": %(max_epochs)i,
},
},
"save_path": "./dae_l2.pkl",
"save_freq": 50
}