Initialize AI hw2 and dataset 1 of 2