diff --git a/cami_src/utils/kolmogorov_smirnoff.py b/cami_src/utils/kolmogorov_smirnoff.py new file mode 100644 index 0000000000000000000000000000000000000000..2341d97f1a2538279cf3bd63b8eadcb60098f241 --- /dev/null +++ b/cami_src/utils/kolmogorov_smirnoff.py @@ -0,0 +1,18 @@ +from scipy.stats import kstest + +def calculate_ks_p_value(data1, data2): + """Calculate the Kolmogorov-Smirnoff p-value for two data sets. + + Parameters + ---------- + data1 : array-like + The first data set. + data2 : array-like + The second data set. + + Returns + ------- + p_value : float + The p-value for the Kolmogorov-Smirnoff test. + """ + return kstest(data1, data2)[1] \ No newline at end of file