32 lines
618 B
Python
32 lines
618 B
Python
|
from pathlib import Path
|
||
|
|
|
||
|
|
defalut_env = {
|
||
|
|
"DATASET_NAME": "rcv1",
|
||
|
|
"DATASET_TARGET": "CCAT",
|
||
|
|
"COMP_ESTIMATORS": [
|
||
|
|
"OUR_BIN_SLD",
|
||
|
|
"OUR_MUL_SLD",
|
||
|
|
"KFCV",
|
||
|
|
"ATC_MC",
|
||
|
|
"ATC_NE",
|
||
|
|
"DOC_FEAT",
|
||
|
|
# "RCA",
|
||
|
|
# "RCA_STAR",
|
||
|
|
],
|
||
|
|
"DATASET_N_PREVS": 9,
|
||
|
|
"OUT_DIR": Path("out"),
|
||
|
|
"PLOT_OUT_DIR": Path("out/plot"),
|
||
|
|
"PROTOCOL_N_PREVS": 21,
|
||
|
|
"PROTOCOL_REPEATS": 100,
|
||
|
|
"SAMPLE_SIZE": 1000,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class Environ:
|
||
|
|
def __init__(self, **kwargs):
|
||
|
|
for k, v in kwargs.items():
|
||
|
|
self.__setattr__(k, v)
|
||
|
|
|
||
|
|
|
||
|
|
env = Environ(**defalut_env)
|