fix doc README and exp006

tks0123456789 · Dec 15, 2016 · 556db06 · 556db06
1 parent cbba1d8
commit 556db06
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -14,11 +14,17 @@ LGBM's max_depth is not the maximum depth of a tree, **maximum depth + 1** !!!
 * LightGBM.ipynb: Modified version of [marugari's work](https://github.com/marugari/Notebooks/blob/ed6aa7835579ce9143850ed5956912895c984d56/LightGBM.ipynb)
 
 * exp004: [Higgs dataset](https://archive.ics.uci.edu/ml/datasets/HIGGS)
-  * n_train              : 10^5, 10^6, 10^7
+  * n_train              : 0.1M, 1M, 10M
   * n_features           : 28
   * max_depth            : 5, 10, 15
 * exp005: Artificial datasets
-  * n_train              : 10^6, 2*10^6
+  * n_train              : 1M, 2M
   * n_features           : 28
   * n_clusters_per_class : 64
   * max_depth            : 5, 10,11,..,16
+* exp006: Artificial datasets, LightGBM only
+  * n_train              : 1M, 2M, 4M
+  * n_features           : 28
+  * max_depth-1          : no limit, log2(num_leaves), 8, 18
+  * num_leaves           : 2**11, 2**13, 2**15
+
diff --git a/exp006.py b/exp006.py
@@ -10,7 +10,7 @@
   n_features          : 28
   n_rounds            : 100
   n_clusters_per_class: 64
-  max_depth-1         : no limit, [11,13,15], 20
+  max_depth-1         : no limit, log2(num_leaves), 8, 18
   num_leaves          : 2**11, 2**13, 2**15
 
                               no_depth_limit  depth  depth_8  depth_18