Julia code for machine learning