Update logreg.md
This commit is contained in:
parent
e0949b7f12
commit
5881947aed
@ -34,13 +34,13 @@ julia> data[1]
|
|||||||
Now we define our model, which will simply be a function from one to the other.
|
Now we define our model, which will simply be a function from one to the other.
|
||||||
|
|
||||||
```julia
|
```julia
|
||||||
model = Chain(
|
m = Chain(
|
||||||
Input(784),
|
Input(784),
|
||||||
Affine(128), relu,
|
Affine(128), relu,
|
||||||
Affine( 64), relu,
|
Affine( 64), relu,
|
||||||
Affine( 10), softmax)
|
Affine( 10), softmax)
|
||||||
|
|
||||||
model = tf(model)
|
model = tf(m)
|
||||||
```
|
```
|
||||||
|
|
||||||
We can try this out on our data already:
|
We can try this out on our data already:
|
||||||
|
Loading…
Reference in New Issue
Block a user