[MRG] Fix plot_unveil_tree_structure.py: incorrect indexing for X_test
@@ -114,11 +114,11 @@ for node_id in node_index:
else:
threshold_sign = ">"
- print("decision id node %s : (X[%s, %s] (= %s) %s %s)"
+ print("decision id node %s : (X_test[%s, %s] (= %s) %s %s)"
% (node_id,
sample_id,
feature[node_id],
- X_test[i, feature[node_id]],
+ X_test[sample_id, feature[node_id]],
threshold_sign,
threshold[node_id]))
