Esempi in Python per load_datasets

Esempio n. 1

File: experiment6.py Progetto: DEKHTIARJonathan/IMA-Deep-Learning

def main(input_directory, num_epochs=5):
    print("Loading data...")
    data = load_datasets(input_directory)

    print("Building model...")
    input_var = theano.tensor.tensor4('inputs')
    target_var = theano.tensor.ivector('targets')
    network = build_network(input_var)
    model = build_model(input_var, target_var, network)

    print("Starting training...")
    train_network(data, model, num_epochs)

    print("\nLearned parameters:")
    params = numpy.array(lasagne.layers.get_all_param_values(network))
    param_file = 'output/{}.npy'.format(str(uuid.uuid4()))
    print("Parameters written to {}".format(param_file))
    params.dump(param_file)

Esempio n. 2

File: experiment5.py Progetto: rchurchley/IMA-Deep-Learning

def main(input_directory, num_epochs=5):
    print("Loading data...")
    data = load_datasets(input_directory)

    print("Building model...")
    input_var = theano.tensor.tensor4('inputs')
    target_var = theano.tensor.ivector('targets')
    network = build_network(input_var)
    model = build_model(input_var, target_var, network)

    print("Starting training...")
    train_network(data, model, num_epochs)

    print("\nLearned parameters:")
    params = numpy.array(lasagne.layers.get_all_param_values(network))
    param_file = 'output/{}.npy'.format(str(uuid.uuid4()))
    print("Parameters written to {}".format(param_file))
    params.dump(param_file)