visible_unit_type: If the visible units are of type binary or gauss
•
main_dir: The main directory where to put the models and the
directories for data and summary
•
model_name: The name of the model used when saving
•
gibbs_sampling_steps: (Optional) Default is 1.
•
learning_rate: (Optional) Set to the default value of 0.01. Specifies
the learning rate.
•
momentum: The value for momentum to use in gradient descent.
Default is 0.9.
•
l2: The l2-weight decay. Default is 0.001.
•
batch_size: (Optional) Default is 10.
•
num_epochs: (Optional) Default is 10.
•
stddev: (Optional) Default is 0.1. Ignored if the visible_unit_type is
not gauss.
•
verbose: (Optional) Default is 0. A value of 1 shows the outputs, and
0 shows nothing.
•
Do'stlaringiz bilan baham: |