Exemplo n.º 1
0
def regression_chicago_crime_weather(p, date_start, date_end, crime_types = None, wards = None):
    crime_data = get_chicago_crime_data(p, date_start, date_end, crime_types, wards)
    weather_data = get_chicago_weather_data(p, date_start, date_end)

    crimes = [x[1] for x in crime_data]
    temps = [x[1] for x in weather_data]
    fill_missing_values(temps)
    
    r = Regression(temps, u"Temperature (°F)", crimes, get_crimes_label(crime_types))    
    r.compute()
    r.plot()