Contains Machine Learning Class Assignment