step_size = 5e-2
if method == 0:
inf_method = approx_inf.EEP(power=1, damping=0.5)
elif method == 1:
inf_method = approx_inf.EEP(power=0.5, damping=0.5)
elif method == 2:
inf_method = approx_inf.EKS(damping=0.5)
elif method == 3:
inf_method = approx_inf.UEP(power=1, damping=0.5)
elif method == 4:
inf_method = approx_inf.UEP(power=0.5, damping=0.5)
elif method == 5:
inf_method = approx_inf.UKS(damping=0.5)
elif method == 6:
inf_method = approx_inf.GHEP(power=1, damping=0.5)
elif method == 7:
inf_method = approx_inf.GHEP(power=0.5, damping=0.5)
elif method == 8:
inf_method = approx_inf.GHKS(damping=0.5)
elif method == 9:
step_size = 1e-2
inf_method = approx_inf.EP(power=1, intmethod='UT', damping=0.05) # this method requies more damping
elif method == 10:
if fold in [1]:
inf_method = approx_inf.EP(power=0.5, intmethod='UT', damping=0.5)
else:
RT = inputs[test, 1:]
YT = Yall[test, :]
if method == 0:
inf_method = approx_inf.EEP(power=1)
elif method == 1:
inf_method = approx_inf.EEP(power=0.5)
elif method == 2:
inf_method = approx_inf.EKS()
elif method == 3:
inf_method = approx_inf.UEP(power=1)
elif method == 4:
inf_method = approx_inf.UEP(power=0.5)
elif method == 5:
inf_method = approx_inf.UKS()
elif method == 6:
inf_method = approx_inf.GHEP(power=1)
elif method == 7:
inf_method = approx_inf.GHEP(power=0.5)
elif method == 8:
inf_method = approx_inf.GHKS()
elif method == 9:
inf_method = approx_inf.EP(power=1, intmethod='UT')
elif method == 10:
inf_method = approx_inf.EP(power=0.5, intmethod='UT')
elif method == 11:
inf_method = approx_inf.EP(power=0.01, intmethod='UT')
