AlexeyAB
2018-08-07 e6c97a53a7b5ac4014d30d236ea2bf5adb4bb521
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);