Don't print non-final tree

msoos · May 23, 2018 · dc2d617 · dc2d617
1 parent 7307016
commit dc2d617
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/learn/mini_pred.py b/scripts/learn/mini_pred.py
@@ -197,7 +197,7 @@ def one_classifier(df, features, to_predict, names, w_name, w_number, final):
     if False:
         calc_cross_val()
 
-    if options.dot is not None:
+    if options.dot is not None and final:
         output_to_dot(clf, features, names[0])
 
     return best_features