Skip to content

Commit

Permalink
Update wl4s_k
Browse files Browse the repository at this point in the history
  • Loading branch information
dongkwan-kim committed Jun 24, 2024
1 parent 0a70ce3 commit ebb9ec0
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions WL4S/wl4s_k.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,27 +5,26 @@
if __name__ == '__main__':
HPARAM_SPACE = {
"stype": ["separated"],
"wl_cumcat": [False],
"wl_cumcat": [False, True],
"hist_norm": [False, True],
}
Cx100 = [8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576]
MORE_HPARAM_SPACE = {
"C": [c / 100 for c in Cx100],
}
DATA_TO_RATIO_SAMPLES = {"HPOMetab": 1800 / 2400, "HPONeuro": 1800 / 4000}
DATA_TO_RATIO_SAMPLES = {"HPOMetab": 1400 / 2400, "HPONeuro": 1400 / 4000}

__args__ = parser.parse_args()
__args__.stype = "separated"
__args__.wl_cumcat = False

MODE = "real_large_precomputation"

if MODE == "syn":
HPARAM_SPACE = {**HPARAM_SPACE, "model": ["LinearSVC"]}
MORE_HPARAM_SPACE = {**MORE_HPARAM_SPACE, "dual": [True, False]}
for k_to_sample in [1, 2, 3, 4]:
for k_to_sample in [None, 1, 2]:
__args__.k_to_sample = k_to_sample
kws = dict(file_dir="../_logs_wl4s_k", log_postfix=f"_{k_to_sample}")
kws = dict(file_dir="../_logs_wl4s_k", log_postfix=f"_{k_to_sample or 0}")
for dataset_name in ["Component", "Density", "Coreness", "CutRatio"]:
__args__.dataset_name = dataset_name
hp_search_for_models(__args__, HPARAM_SPACE, MORE_HPARAM_SPACE, **kws)
Expand Down

0 comments on commit ebb9ec0

Please sign in to comment.