AlexeyAB
2018-04-03 ee8a9419227d551f9c1917dec843d3c4dd526a66
src/tree.h
@@ -5,6 +5,7 @@
    int *leaf;
    int n;
    int *parent;
   int *child;
    int *group;
    char **name;
@@ -14,6 +15,7 @@
} 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);