hats5

examples/benchmarks/CatBoost/workflow_config_catboost.yaml

@@ -32,7 +32,11 @@ task:
             loss: RMSE
             learning_rate: 0.0421
             subsample: 0.8789
+            max_depth: 6
+            num_leaves: 100
             thread_count: 20
+            grow_policy: Lossguide
+            boostrap_type: Poisson
     dataset:
         class: DatasetH
         module_path: qlib.data.dataset

examples/run_all_model.py

@@ -178,7 +178,6 @@ def get_all_folders() -> dict:
     folders = dict()
     for f in os.scandir("benchmarks"):
         path = Path("benchmarks") / f.name
-        if f.name != "TFT":
         folders[f.name] = str(path.resolve())
     return folders
 

qlib/data/dataset/loader.py

@@ -21,18 +21,16 @@ class DataLoader(abc.ABC):
     @abc.abstractmethod
     def load(self, instruments, start_time=None, end_time=None) -> pd.DataFrame:
         """
-        load the data as pd.DataFrame
+        load the data as pd.DataFrame.
 
         Parameters
         ----------
-        self : [TODO:type]
-            [TODO:description]
-        instruments : [TODO:type]
-            [TODO:description]
-        start_time : [TODO:type]
-            [TODO:description]
-        end_time : [TODO:type]
-            [TODO:description]
+        instruments : str or dict
+            it can either be the market name or the config file of instruments generated by InstrumentProvider.
+        start_time : str
+            start of the time range.
+        end_time : str
+            end of the time range.
 
         Returns
         -------