ml_project_hw2_inference