chrisvnicholson / jutsu.ai

Clojure wrapper for deeplearning4j

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

jutsu.ai

Clojure wrapper for deeplearning4j with some added syntactic sugar.

What if I told you that you could do machine learning on the JVM without wanting to cry or set your hair on fire? The goal of this library is to be the most syntactically elegant machine learning library for Clojure/Java ecosystem. jutsu.ai uses Clojure specific code idioms, like Data as Code, to create an aesthetic and declarative api.

jutsu.ai makes use of my laziest but cleverest achievement which is that it the network configuration translates clojure keywords into java method calls. The implication of this is that the majority of the config options provided by deeplearning4j are accessible with this library.

The best of all, no more snakes!

alt text

Currently this project is a work in progress. However, no drastic API changes are expected. Configurations have been tested on basic feedforward nets, RNN's, and CNN's. If you can't seem to find a configuration feature you are looking for submit an issue and I'll do my best to help.

If you have any suggestions or fixes please feel free to submit a PR as well!

Currently need help documenting the different configuration options. However, deeplearning4j has a very good resource that I recommend checking out.

Usage

Install jutsu.ai by including this in your project dependencies:

[hswick/jutsu.ai "0.1.0"]

You will also need to include a ND4J backend:

For CPU:

[org.nd4j/nd4j-native-platform "0.8.0"]

For GPU:

[org.nd4j/nd4j-cuda-8.0-platform "0.8.0"]

Here is a toy example demonstrating classifying the well known iris dataset.

(ns skynet.core
  (:require [jutsu.ai.core :as ai]))

;;Example configuration of a simple multilayer feedforward neural network architecture
;;classifying the iris dataset.
(def iris-net-config
	[:layers [[:dense [:n-in 4 :n-out 10 :activation :relu]]
			  [:dense [:n-in 10 :n-out 10 :activation :relu]]
			  [:output :negative-log-likelihood [:n-in 10 :n-out 3 :activation :softmax]]]])

(def iris-net
	(-> iris-net-config
	ai/network-config
	ai/initialize-net))

(defn iris-train []
  (let [iris-iterator (ai/classification-csv-iterator "iris.csv" 150 4 3)]
    (-> iris-net
	(ai/train-net! 200 iris-iterator)
	(ai/save-model "iris-model"))
  (.reset iris-iterator)
  (println (ai/evaluate iris-net iris-iterator))))

(iris-train)

Make sure to take a look at the wiki for more documentation on the configuration and other parts of the setup: wiki

If you want to use a GUI to visualize training progress check out this plugin I made: jutsu.ai.ui

Here is a video tutorial I made demonstrating the use of jutsu.ai.

Dev

Run boot night to startup nightlight and begin editing the project in a browser.

License

Distributed under the Eclipse Public License either version 1.0 or (at your option) any later version.

About

Clojure wrapper for deeplearning4j

License:Eclipse Public License 1.0


Languages

Language:Clojure 100.0%