示例#1
0
文件: sim.py 项目: weipeng/pyepi
def sim_sir_filtered():
    params = read_params('./data/params/params.csv')
    params['filtering'] = True
    sir = SIR(params)

    sir.predict_with_filter()

    pprint(sir.scores)
    return sir.score
示例#2
0
文件: sim.py 项目: sdwfrost/pyepi
def sim_sir_filtered():
    params = read_params('./data/params/params.csv')
    params['filtering'] = True
    sir = SIR(params)

    sir.predict_with_filter()

    pprint(sir.scores)
    return sir.score
示例#3
0
def sim_psir_filtered(ens, year, date, params=None):
    if not params:
        params = read_params('./data/params/params%s.csv' % year)
    params['filtering'] = True
    params['time_varying'] = False
    sir = ParticleSIR(ens, params)

    sir.predict_with_filter()

    out_str = ','.join(map(str, sir.IS))
    pprint(sir.scores)
    path = '%s_%s_predictions_example' % (date[0], date[1])
    write_file(path, year, sir, out_str)
示例#4
0
文件: sim.py 项目: sdwfrost/pyepi
def sim_easir_filtered(ens, year, cov_type, params):
    if not params:
        params = read_params('./data/params/params%s.csv' % year)
    params['filtering'] = True
    sir = EnsembleAdjustmentSIR(ens, params)

    sir.filter.cov_type = cov_type
    sir.predict_with_filter()

    out_str = ','.join(map(str, sir.IS))
    path = 'centered_eakf' if cov_type == 'c' else 'uncentered_eakf'
    write_file(path, year, sir, out_str)
    return sir.score
示例#5
0
文件: sim.py 项目: weipeng/pyepi
def sim_psir_filtered(ens, year, params=None):
    if not params:
        params = read_params('./data/params/params%s.csv' % year)
    params['filtering'] = True
    params['time_varying'] = False
    sir = ParticleSIR(ens, params)

    sir.predict_with_filter()
 
    out_str = ','.join(map(str, sir.IS))
    pprint(sir.scores)
    path = 'centered_pkf'
    write_file(path, year, sir, out_str)
示例#6
0
文件: sim.py 项目: weipeng/pyepi
def sim_easir_filtered(ens, year, cov_type, params):
    if not params:
        params = read_params('./data/params/params%s.csv' % year)
    params['filtering'] = True
    sir = EnsembleAdjustmentSIR(ens, params)

    sir.filter.cov_type = cov_type
    sir.predict_with_filter()
 
    out_str = ','.join(map(str, sir.IS))
    path = 'centered_eakf' if cov_type == 'c' else 'uncentered_eakf'
    write_file(path, year, sir, out_str)
    return sir.score
示例#7
0
文件: sim.py 项目: sdwfrost/pyepi
def sim_basssir_filtered(ens, year, cov_type, params=None):
    if not params:
        params = read_params('./data/params/params%s.csv' % year)
    params['filtering'] = True
    params['time_varying'] = False
    sir = BassSIR(ens, params)

    sir.err_bnd = 0.00001
    sir.filter.cov_type = cov_type
    sir.predict_with_filter()

    out_str = ','.join(map(str, sir.IS))
    path = 'centered_bass' if cov_type == 'c' else 'uncentered_bass'

    pprint(sir.scores)
    write_file(path, year, sir, out_str)
    return sir.score
示例#8
0
文件: sim.py 项目: weipeng/pyepi
def sim_basssir_filtered(ens, year, cov_type, params=None):
    if not params:
        params = read_params('./data/params/params%s.csv' % year)
    params['filtering'] = True
    params['time_varying'] = False
    sir = BassSIR(ens, params)

    sir.err_bnd = 0.00001
    sir.filter.cov_type = cov_type
    sir.predict_with_filter()
 
    out_str = ','.join(map(str, sir.IS))
    path = 'centered_bass' if cov_type == 'c' else 'uncentered_bass'

    pprint(sir.scores)
    write_file(path, year, sir, out_str)
    return sir.score
示例#9
0
文件: sim.py 项目: sdwfrost/pyepi
def sim_sir():
    params = read_params('./data/params/params2013-14-sir.csv')
    sir = SIR(params)

    sir.predict()
    return sir.scores
示例#10
0
文件: sim.py 项目: weipeng/pyepi
def sim_sir():
    params = read_params('./data/params/params2013-14-sir.csv')
    sir = SIR(params)

    sir.predict()
    return sir.scores