src/tree.h
@@ -5,6 +5,7 @@ int *leaf; int n; int *parent; int *child; int *group; char **name; @@ -14,7 +15,9 @@ } tree; tree *read_tree(char *filename); int hierarchy_top_prediction(float *predictions, tree *hier, float thresh, int stride); void hierarchy_predictions(float *predictions, int n, tree *hier, int only_leaves); void change_leaves(tree *t, char *leaf_list); float get_hierarchy_probability(float *x, tree *hier, int c); #endif
