Category prediction task for CSE158. See course homepage for assignment spec.