epoch: 0, accuracy: 2.8571

model.fut

@@ -45,7 +45,7 @@ entry specs =
 let nn = model ()
 
 entry predict = \ws inputs ->
-  dl.nn.predict (model ()) ws inputs (dl.nn.softmax 3) |> map dl.nn.argmax
+  dl.nn.predict (model ()) ws inputs (dl.nn.identity 3)
 --  let nn = model ()
 --  in dl.nn.predict nn ws inputs (dl.nn.softmax 10) |> map dl.nn.argmax