Wrangle the data frame df that we read in at the top of the notebook into the format required above, saving it as variable df2. Separate out the dependent variable :students_infected (which you will also have to wrangle into existance) and save it as variable yy to use for training.

room_number lecturer_mask box_filter good_ventilation student_number studen 1.0 0.0 1.0 5.0 2 1.0 0.0 1.0 2 5.0 3 1.0 0.0 1.0 5.0 4 1.0 0.0 1.0 5.0 5 1.0 0.0 1.0 5.0 6 1 1.0 0.0 1.0 5.0 7 1.0 0.0 1.0 5.0 8 1.0 0.0 1.0 10.0 9 1.0 0.0 1.0 CO 10.0 10 1 1.0 0.0 1.0 10 10.0 i more 69300 1100 0.0 1.0 1.0 63 45.0 df