5,443
个编辑
更改
→计算EI的源代码
q = m
q = m
# replace 0 to a small positive number to avoid log error
# replace 0 to a positive number to avoid log error
eps = 1E-10
eps = 0.5
tpm_e = np.where(tpm==0, eps, tpm)
tpm_e = np.where(tpm==0, eps, tpm)
puy_e = np.where(puy==0, eps, puy)
puy_e = np.where(puy==0, eps, puy)
deg = deg / np.log2(log_base)
deg = deg / np.log2(log_base)
det_c = det / np.log2(q)
det_c = det / np.log2(q) * np.log2(log_base)
deg_c = deg / np.log2(q)
deg_c = deg / np.log2(q) * np.log2(log_base)
# calculate total EI
# calculate total EI
ei_all = ei_x.mean()
ei_all = ei_x.mean()
eff = ei_all / np.log2(q)
eff = ei_all / np.log2(q) * np.log2(log_base)
return ei_all,det,deg,eff,det_c,deg_c
return ei_all,det,deg,eff,det_c,deg_c
tpm_ei(mi_states)
</syntaxhighlight>
</syntaxhighlight>