Azure DevOps Demo Models · main

enrollment_forecast.py

education/university-enrollment-forecast/src/enrollment_forecast.py · 644 bytes · read-only public source view

from __future__ import annotations
import csv

enrollment=28600
applications=52000
admit_rate=0.48
yield_rate=0.29
retention=0.88
with open('enrollment_forecast.csv','w',newline='',encoding='utf-8') as handle:
    writer=csv.writer(handle)
    writer.writerow(['year','applications','new_students','continuing_students','total_enrollment'])
    for year in range(1,9):
        new_students=applications*admit_rate*yield_rate
        continuing=enrollment*retention
        enrollment=new_students+continuing
        writer.writerow([year,round(applications),round(new_students),round(continuing),round(enrollment)])
        applications*=1.012