Machine Learning in Elixir: Model output "iris_features" => nil, not tuple

[adamt]

@seanmor5

Title: Machine Learning in Elixir (epub so pages different i think)

From book:

Current output:

Diff:

  inputs: %{"iris_features" => nil}

[seanmor5]

Thanks for pointing this out! To get the {nil, 4} you can specify a shape in the input:

Axon.input("iris_features", shape: {nil, 4})
|> Axon.dense(3, activation: :softmax)