ht/spacing update

h2oai · Jan 18, 2024 · f1341d7 · f1341d7
1 parent bd1ea84
commit f1341d7
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 8 deletions.
diff --git a/h2o-bindings/bin/custom/python/gen_dt.py b/h2o-bindings/bin/custom/python/gen_dt.py
@@ -15,7 +15,9 @@
     >>> target_variable = 'CAPSULE'
     >>> prostate[target_variable] = prostate[target_variable].asfactor()
     >>> train, test = prostate.split_frame(ratios=[0.7])
-    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, categorical_encoding="auto")
+    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+    ...                                    max_depth=5,
+    ...                                    categorical_encoding="auto")
     >>> sdt_h2o.train(y=target_variable, training_frame=train)
     >>> pred_test = sdt_h2o.predict(test)
     """,
@@ -27,7 +29,9 @@
     >>> target_variable = 'CAPSULE'
     >>> prostate[target_variable] = prostate[target_variable].asfactor()
     >>> train, test = prostate.split_frame(ratios=[0.7])
-    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, ignore_const_cols=True)
+    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+    ...                                    max_depth=5,
+    ...                                    ignore_const_cols=True)
     >>> sdt_h2o.train(y=target_variable, training_frame=train)
     >>> pred_test = sdt_h2o.predict(test)
     """,
@@ -39,7 +43,9 @@
     >>> target_variable = 'CAPSULE'
     >>> prostate[target_variable] = prostate[target_variable].asfactor()
     >>> train, test = prostate.split_frame(ratios=[0.7])
-    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, max_depth=20)
+    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+    ...                                    max_depth=5,
+    ...                                    max_depth=20)
     >>> sdt_h2o.train(y=target_variable, training_frame=train)
     >>> pred_test = sdt_h2o.predict(test)
     """,
@@ -51,7 +57,9 @@
     >>> target_variable = 'CAPSULE'
     >>> prostate[target_variable] = prostate[target_variable].asfactor()
     >>> train, test = prostate.split_frame(ratios=[0.7])
-    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, min_rows=10)
+    >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+    ...                                    max_depth=5,
+    ...                                    min_rows=10)
     >>> sdt_h2o.train(y=target_variable, training_frame=train)
     >>> pred_test = sdt_h2o.predict(test)
     """

diff --git a/h2o-py/h2o/estimators/decision_tree.py b/h2o-py/h2o/estimators/decision_tree.py
@@ -117,7 +117,9 @@ def ignore_const_cols(self):
         >>> target_variable = 'CAPSULE'
         >>> prostate[target_variable] = prostate[target_variable].asfactor()
         >>> train, test = prostate.split_frame(ratios=[0.7])
-        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, ignore_const_cols=True)
+        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+        ...                                    max_depth=5,
+        ...                                    ignore_const_cols=True)
         >>> sdt_h2o.train(y=target_variable, training_frame=train)
         >>> pred_test = sdt_h2o.predict(test)
         """
@@ -145,7 +147,9 @@ def categorical_encoding(self):
         >>> target_variable = 'CAPSULE'
         >>> prostate[target_variable] = prostate[target_variable].asfactor()
         >>> train, test = prostate.split_frame(ratios=[0.7])
-        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, categorical_encoding="auto")
+        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+        ...                                    max_depth=5,
+        ...                                    categorical_encoding="auto")
         >>> sdt_h2o.train(y=target_variable, training_frame=train)
         >>> pred_test = sdt_h2o.predict(test)
         """
@@ -200,7 +204,9 @@ def max_depth(self):
         >>> target_variable = 'CAPSULE'
         >>> prostate[target_variable] = prostate[target_variable].asfactor()
         >>> train, test = prostate.split_frame(ratios=[0.7])
-        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, max_depth=20)
+        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+        ...                                    max_depth=5,
+        ...                                    max_depth=20)
         >>> sdt_h2o.train(y=target_variable, training_frame=train)
         >>> pred_test = sdt_h2o.predict(test)
         """
@@ -227,7 +233,9 @@ def min_rows(self):
         >>> target_variable = 'CAPSULE'
         >>> prostate[target_variable] = prostate[target_variable].asfactor()
         >>> train, test = prostate.split_frame(ratios=[0.7])
-        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex", max_depth=5, min_rows=10)
+        >>> sdt_h2o = H2ODecisionTreeEstimator(model_id="decision_tree.hex",
+        ...                                    max_depth=5,
+        ...                                    min_rows=10)
         >>> sdt_h2o.train(y=target_variable, training_frame=train)
         >>> pred_test = sdt_h2o.predict(test)
         """