Update dtc.py

mlpack · Apr 5, 2017 · 9c70c11 · 9c70c11
1 parent f86b79a
commit 9c70c11
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/methods/scikit/dtc.py b/methods/scikit/dtc.py
@@ -73,7 +73,8 @@ def BuildModel(self, data, labels):
                                  max_depth=self.max_depth,
                                  random_state=self.seed,
                                  splitter = self.splitter,
-                                 min_samples_split = self.min_samples_split,                                 min_weight_fraction_leaf=self.min_weight_fraction_leaf,
+                                 min_samples_split = self.min_samples_split,                                 
+                                 min_weight_fraction_leaf=self.min_weight_fraction_leaf,
                                  max_features = self.max_features,
                                  max_leaf_nodes = self.max_leaf_nodes,
                                  min_impurity_split = self.min_impurity_split,
@@ -105,6 +106,7 @@ def RunDTCScikit(q):
       msl = re.search("--min_samples_leaf (\d+)", options)
       mf = re.search("--max_features (\d+)", options)
       mln = re.search("--max_leaf_nodes (\d+)", options)
+
       self.criterion = 'gini' if not c else str(c.group(1))
       self.max_depth = None if not d else int(d.group(1))
       self.splitter = 'best' if not s else str(s.group(1))