You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
548 B
22 lines
548 B
6 years ago
|
import numpy as np
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def gen_chi2_ppf_lookup(max_dim=200):
|
||
|
|
from scipy.stats import chi2
|
||
|
|
table = np.zeros((max_dim, 98))
|
||
|
|
for dim in range(1,max_dim):
|
||
|
|
table[dim] = chi2.ppf(np.arange(.01, .99, .01), dim)
|
||
|
|
#outfile = open('chi2_lookup_table', 'w')
|
||
|
|
np.save('chi2_lookup_table', table)
|
||
|
|
|
||
|
|
|
||
|
|
def chi2_ppf(p, dim):
|
||
|
|
table = np.load(os.path.dirname(os.path.realpath(__file__)) + '/chi2_lookup_table.npy')
|
||
|
|
result = np.interp(p, np.arange(.01, .99, .01), table[dim])
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
if __name__== "__main__":
|
||
|
|
gen_chi2_ppf_lookup()
|