/aorun

Deep Learning over PyTorch

Primary LanguagePythonMIT LicenseMIT

Aorun: Deep Learning over PyTorch

build coverage Code Climate python license

Aorun intend to implement an API similar to 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.predict(X_test)

Install

First of all, it's import to mention that this project is develop with Python 3.5+ in mind. I do not recommend using Aorun with older versions.

As prerequisite, you have to install the latest stable version of PyTorch

Then you can install aorun as any other python package, with pip:

$ pip install aorun

Why Aorun?

Aorun is a Dinosaur. Dinosaurs are cool.