Skip to main content

A neural network library built on top of TensorFlow for quickly building deep learning models.

Project description

A neural network library built on top of TensorFlow for quickly building deep learning models.

Installation

pip install nn

Example

import nn

# Create the model
@nn.model
def model(inputs):
    # Define the network architecture (layers, number of units, activations)
    hidden = nn.Dense(units=64, activation='relu')(inputs)
    outputs = nn.Dense(units=10)(hidden)

    # Configure the learning process (loss, optimizer, evaluation metrics)
    return dict(outputs=outputs,
                loss='softmax_cross_entropy',
                optimizer=('GradientDescent', 0.001),
                metrics=['accuracy'])

# Train the model using training data:
model.train(x_train, y_train, epochs=30, batch_size=128)

# Evaluate the model performance on test or validation data:
loss_and_metrics = model.evaluate(x_test, y_test)

# Use the model to make predictions for new data:
predictions = model.predict(x)

Documentation

See documentation.

License

MIT

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
nn-0.0.9.tar.gz (7.5 kB) Copy SHA256 hash SHA256 Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page