Commit eef7fc6b authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add n_jobs and backend to dm.quickstart

parent e66ed791
...@@ -509,7 +509,7 @@ class CsstMsDataManager: ...@@ -509,7 +509,7 @@ class CsstMsDataManager:
@staticmethod @staticmethod
def quickstart(ver_sim="C5.2", datatype="mbi", dir_l1=".", exposure_id=100, def quickstart(ver_sim="C5.2", datatype="mbi", dir_l1=".", exposure_id=100,
use_dfs=False, dfs_node="kmust", clear_dir=False): use_dfs=False, dfs_node="kmust", clear_l1=False, n_jobs=18, backend="multiprocessing"):
""" """
Quick dataset generator for tests on dandelion or PML Quick dataset generator for tests on dandelion or PML
...@@ -527,8 +527,12 @@ class CsstMsDataManager: ...@@ -527,8 +527,12 @@ class CsstMsDataManager:
If True, use DFS. If True, use DFS.
dfs_node : str dfs_node : str
The DFS node. Defaults to "kmust", could be "pml". The DFS node. Defaults to "kmust", could be "pml".
clear_dir : bool clear_l1 : bool
If True, clear dir_l1. If True, clear dir_l1.
n_jobs : int
The number of jobs.
backend : str
The joblib backend.
Returns Returns
------- -------
...@@ -564,7 +568,7 @@ class CsstMsDataManager: ...@@ -564,7 +568,7 @@ class CsstMsDataManager:
return CsstMsDataManager.from_dir( return CsstMsDataManager.from_dir(
ver_sim=ver_sim, datatype=datatype, dir_l0=dir_l0, dir_l1=dir_l1, path_aux=path_aux, ver_sim=ver_sim, datatype=datatype, dir_l0=dir_l0, dir_l1=dir_l1, path_aux=path_aux,
use_dfs=use_dfs, dfs_node=dfs_node) use_dfs=use_dfs, dfs_node=dfs_node, n_jobs=n_jobs, backend=backend)
def __repr__(self): def __repr__(self):
lines = "" lines = ""
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment