Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/ramon-oliveira/aorun

Deep Learning over PyTorch
https://github.com/ramon-oliveira/aorun

bayesian-inference deeplearning neural-networks pytorch

Last synced: about 1 month ago
JSON representation

Deep Learning over PyTorch

Awesome Lists containing this project

README

        

# Aorun: Deep Learning over PyTorch

[![build](https://travis-ci.org/ramon-oliveira/aorun.svg?branch=master)](https://travis-ci.org/ramon-oliveira/aorun)
[![coverage](https://coveralls.io/repos/github/ramon-oliveira/aorun/badge.svg)](https://coveralls.io/github/ramon-oliveira/aorun)
[![Code Climate](https://codeclimate.com/github/ramon-oliveira/aorun/badges/gpa.svg)](https://codeclimate.com/github/ramon-oliveira/aorun)
[![python](https://img.shields.io/pypi/pyversions/aorun.svg)](https://pypi.python.org/pypi/aorun)
[![license](https://img.shields.io/github/license/ramon-oliveira/aorun.svg)](https://github.com/ramon-oliveira/aorun/blob/master/LICENSE)

Aorun intend to implement an API similar to [Keras](https://keras.io) with PyTorch as backend.

## Getting started

Here is a simple regression example:

```python
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](http://pytorch.org)

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

## Why Aorun?

[Aorun](https://en.wikipedia.org/wiki/Aorun) is a Dinosaur. Dinosaurs are cool.