Skip to content

benglard/symnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

symnn

Neural Networks for symtorch.

As much as symtorch is still a work-in-progress, symnn is a work-in-progress even more so.

TODO:

  • more training/optim methods
  • more layers
  • preprocessing?
  • more examples
  • even more

Example Usage

require 'symnn'
local model = symnn.Sequential()
model:add(symnn.Reshape(784))
model:add(symnn.Linear(784, 100))
model:add(symnn.Sigmoid())
model:add(symnn.Linear(100, 10))
model:add(symnn.SoftMax())
model:add(symnn.ClassNLL())
print(model)

local x = symtorch.Tensor(28, 28):rand()
local y = model:forward(x)
local target = 5
local cost = model:backward(target)

Installation

> (sudo) luarocks install https://raw.githubusercontent.com/benglard/luaclass/master/luaclass-scm-1.rockspec
> (sudo) luarocks install https://raw.githubusercontent.com/benglard/luaimport/master/luaimport-scm-1.rockspec
> (sudo) luarocks install https://raw.githubusercontent.com/benglard/symtorch/master/symtorch-scm-1.rockspec
> (sudo) luarocks install https://raw.githubusercontent.com/benglard/symnn/master/symnn-scm-1.rockspec

About

Neural networks for symtorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages