aorun

Deep Learning over PyTorch


Keywords
neural-networks, deep-learning, pytorch, bayesian-inference, deeplearning
License
MIT
Install
pip install aorun==0.1

Documentation

Aorun: Deep Learning over PyTorch

build coverage python license

Aorun intend to be a Keras with PyTorch as backend.

Getting started

Here is a simple regression example:

from aorun.models import Model
from aorun.layers import Dense, Activation

model = Model()
model.add(Dense(10, input_dim=3))
model.add(Activation('relu'))
model.add(Dense(1))

model.fit(X_train, y_train, loss='mse', optimizer='adam')

y_pred = model.forward(X_test)

Why Aorun?

Aorun is a Dinosaur. Dinosaurs are cool.