Machine Learning in Elixir: Page 130 undefined function error

@seanmor5

Use of Nx.Random.uniform_split/2 I think should be Nx.Random.uniform_split/4, e.g.:

Nx.Random.uniform_split(new_key, 0, 1, shape: {})
|> NeuralNetwork.predict(w1, b1, w2, b2)