DOC correct shape of Tree.value

scikit-learn · Aug 14, 2014 · 6d8ccbc · 6d8ccbc
1 parent 5202312
commit 6d8ccbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklearn/tree/_tree.pyx b/sklearn/tree/_tree.pyx
@@ -2202,7 +2202,7 @@ cdef class Tree:
     threshold : array of double, shape [node_count]
         threshold[i] holds the threshold for the internal node i.
 
-    value : array of double, shape [node_count, n_outputs, ]
+    value : array of double, shape [node_count, n_outputs, max_n_classes]
         Contains the constant prediction value of each node.
 
     impurity : array of double, shape [node_count]