Skip to content

Commit

Permalink
black formatted proglearn
Browse files Browse the repository at this point in the history
  • Loading branch information
EYezerets committed Oct 21, 2020
1 parent 52409a9 commit c262c6b
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions proglearn/forest.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,7 +244,7 @@ class UncertaintyForest:
max_depth : int, default=30
The maximum depth of a tree in the UncertaintyForest
tree_construction_proportion : float, default = 0.67
The proportions of the input data set aside to train each decision
tree. The remainder of the data is used to fill in voting posteriors.
Expand All @@ -256,7 +256,13 @@ class UncertaintyForest:
inference.
"""

def __init__(self, n_estimators=100, finite_sample_correction=False, max_depth=30, tree_construction_proportion=0.67):
def __init__(
self,
n_estimators=100,
finite_sample_correction=False,
max_depth=30,
tree_construction_proportion=0.67,
):
self.n_estimators = n_estimators
self.finite_sample_correction = finite_sample_correction
self.max_depth = max_depth
Expand Down

0 comments on commit c262c6b

Please sign in to comment.