_ff.py 文件源码

def _init_parser(parser):
  """Initializes the parser for feed-forward models.
  """
  optimization = parser.add_argument_group(title='Optimization',
    description='Arguments determining the optimizer behavior.')
  optimization.add_argument('--learning-rate', metavar='rate', type=float, default=0.01,
                            help='The magnitude of learning to perform at each step.')

  nn = parser.add_argument_group(title='Neural Network',
    description='Arguments controlling the structure of the neural network.')
  nn.add_argument('--hidden-layers', metavar='units', type=int, required=False,
                  action=parser.var_args_action,
                  help='The size of each hidden layer to add.')