nn 0.0.4

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

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

Installation

pip install nn

Example

import nn

# Define the network (layers, number of units, activations) as a function:
def network(inputs):
    hidden = nn.Dense(units=64, activation='relu')(inputs)
    outputs = nn.Dense(units=10)(hidden)
    return outputs

# Create a model by configuring its learning process (loss, optimizer, evaluation metrics):
model = nn.Model(network,
                 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