using PlutoUI using CSV, DataFrames using Zygote using LinearAlgebra, Statistics using Plots using RandomIn julia. your task is to create a function called predict_transmission that takes as an argument a data frame with the following columns: :room_number: The room ID (Type: Float64, values: 1-1,100) :lecturer_mask: Whether the professor was wearing a mask (Type: Float64, values: 1 or 0) :box_filter: Whether the room had a supplemental air filter operating (Type: Float64, values: 1 or 0) :good_ventilation: Whether the room has an upgraded HVAC system with filtered air (Type: Float64, values: 1 or 0) :student_mask_frac: The fraction of students wearing a mask (Type: Float64, values: between 0 and 1) :student_vaccine_frac: The fraction of students that have been vaccinated (Type: Float64, values: between 0 and 1)