Skip to content

Commit

Permalink
Fix parameters in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
achamma723 committed Jul 10, 2024
1 parent f9c2830 commit 577a9c7
Show file tree
Hide file tree
Showing 2 changed files with 34 additions and 34 deletions.
6 changes: 3 additions & 3 deletions doc_conf/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,9 @@
bibtex_footbibliography_header = ""

# General information about the project.
project = u'HiDimStat'
copyright = u'2024, Mind-Inria'
author = u'Mind-Inria'
project = "HiDimStat"
copyright = "2024, Mind-Inria"
author = "Mind-Inria"

# The version info for the project you're documenting, acts as replacement for
# |version| and |release|, also used in various other places throughout the
Expand Down
62 changes: 31 additions & 31 deletions hidimstat/test/test_BBI.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,10 @@


def _generate_data(
n_samples=100, n_features=10, prob_type="regression", grps_exp=False, seed=2024
n_samples=100, n_features=10, problem_type="regression", grps_exp=False, seed=2024
):

if prob_type == "regression":
if problem_type == "regression":
X, y = make_regression(
n_samples=n_samples,
noise=0.2,
Expand Down Expand Up @@ -60,16 +60,16 @@ def _generate_data(

def test_BBI_reg():

X, y, _, list_nominal = _generate_data(prob_type="regression")
X, y, _, list_nominal = _generate_data(problem_type="regression")
# DNN
bbi_reg_dnn = BlockBasedImportance(
estimator=None,
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
group_stacking=False,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -84,11 +84,11 @@ def test_BBI_reg():
bbi_reg_rf = BlockBasedImportance(
estimator="RF",
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
group_stacking=False,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -102,16 +102,16 @@ def test_BBI_reg():

def test_BBI_class():

X, y, _, list_nominal = _generate_data(prob_type="classification")
X, y, _, list_nominal = _generate_data(problem_type="classification")
# DNN
bbi_class_dnn = BlockBasedImportance(
estimator=None,
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
group_stacking=False,
prob_type="classification",
problem_type="classification",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -126,11 +126,11 @@ def test_BBI_class():
bbi_class_rf = BlockBasedImportance(
estimator="RF",
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
group_stacking=False,
prob_type="classification",
problem_type="classification",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -149,10 +149,10 @@ def test_BBI_condDNN():
bbi_res = BlockBasedImportance(
estimator=None,
importance_estimator=None,
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
group_stacking=False,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -167,10 +167,10 @@ def test_BBI_condDNN():
bbi_samp = BlockBasedImportance(
estimator=None,
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
group_stacking=False,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -188,11 +188,11 @@ def test_BBI_permDNN():
bbi_perm = BlockBasedImportance(
estimator=None,
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
group_stacking=False,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -211,12 +211,12 @@ def test_BBI_grp():
bbi_grp_noStack = BlockBasedImportance(
estimator="RF",
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
groups=grps,
group_stacking=False,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand All @@ -231,12 +231,12 @@ def test_BBI_grp():
bbi_grp_stack = BlockBasedImportance(
estimator="RF",
importance_estimator="Mod_RF",
do_hyper=True,
dict_hyper=None,
do_hypertuning=True,
dict_hypertuning=None,
conditional=False,
groups=grps,
group_stacking=True,
prob_type="regression",
problem_type="regression",
k_fold=2,
list_nominal=list_nominal,
n_jobs=10,
Expand Down

0 comments on commit 577a9c7

Please sign in to comment.