Brian Broll
c96606fccc
WIP #26 Removed old dimensionality file WIP #26 Updated default calculateDimensionality attr WIP #26 Added detection of inferred layer (if not already set) Updated devTests and tests. WIP #26 WIP #26 Updated nn library
14 linhas
334 B
Lua
14 linhas
334 B
Lua
require 'nn'
|
|
|
|
model = nn.Sequential()
|
|
model:add(nn.Reshape(100))
|
|
model:add(nn.Linear(100, 300))
|
|
model:add(nn.RReLU())
|
|
model:add(nn.Linear(300, 100))
|
|
model:add(nn.ReLU())
|
|
model:add(nn.Linear(100, 100))
|
|
model:add(nn.Sigmoid())
|
|
model:add(nn.Linear(100, 120))
|
|
model:add(nn.LeakyReLU())
|
|
model:add(nn.Linear(120, 5))
|
|
model:add(nn.SoftMax()) |