Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Fast kmerf #369

Merged
merged 13 commits into from
May 24, 2023
Prev Previous commit
Next Next commit
fix build errors
  • Loading branch information
sampan501 committed May 17, 2023
commit 8a343c39a45b056e350d3f5984671274ab27abdb
4 changes: 2 additions & 2 deletions hyppo/ksample/ksamp.py
Original file line number Diff line number Diff line change
Expand Up @@ -184,7 +184,7 @@ def __init__(self, indep_test, compute_distkern="euclidean", bias=False, **kwarg
"hsic": {"bias": bias, "compute_kernel": compute_distkern},
"hhg": {"compute_distance": compute_distkern},
"mgc": {"compute_distance": compute_distkern},
"kmerf": {"forest": "classifier"},
"kmerf": {"forest": "classifier", "is_ksamp": k_sample_transform},
"rv": {},
"cca": {},
}
Expand Down Expand Up @@ -299,7 +299,7 @@ def test(self, *args, reps=1000, workers=1, auto=True, random_state=None):
if self.indep_test_name in ["dcorr", "hsic"]:
kwargs = {"auto": auto}
elif self.indep_test_name in ["kmerf"]:
kwargs = {"auto": auto, "is_ksamp": k_sample_transform}
kwargs = {"auto": auto}

return self.indep_test.test(
u, v, reps, workers, **kwargs, random_state=random_state
Expand Down