Example #1
0
def create_training_setup(config,
                          input_dim=DEFAULT_INPUT_DIM,
                          output_dim=DEFAULT_OUTPUT_DIM):
    # create setup from configuration
    print_subtitle("creating setup")
    hidden_width, num_layers, max_iterations, learn_rate, activation_fn = config
    print "Configured!"
    # make neural network
    neural_network = NeuralNetwork.NeuralNetwork(input_dim, output_dim,
                                                 hidden_width, num_layers,
                                                 activation_fn)
    print "Neural network initialized!"
    # make a trainer
    trainer = Trainer.Backpropagation(max_iterations=max_iterations,
                                      learn_rate=learn_rate)
    print "Trainer initialized!"

    return neural_network, trainer