For any randos that stumble on this gist, you should consider buying Machine Learning in Elixir. Despite any notes I add here, it's an exceptional book that's worth your time and money.
I can't get the code snippet to work as written:
cols = ~w(sepal_width sepal_length petal_length petal_width)