From cd8a3dcb4ca42f22ad8f46a95e00977c92be6bbd Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 08 Feb 2018 23:22:42 +0000
Subject: [PATCH] Compile fixes
---
src/tree.c | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/src/tree.c b/src/tree.c
index cd9fcd1..dfa4178 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -24,6 +24,16 @@
fprintf(stderr, "Found %d leaves.\n", found);
}
+float get_hierarchy_probability(float *x, tree *hier, int c)
+{
+ float p = 1;
+ while(c >= 0){
+ p = p * x[c];
+ c = hier->parent[c];
+ }
+ return p;
+}
+
void hierarchy_predictions(float *predictions, int n, tree *hier, int only_leaves)
{
int j;
--
Gitblit v1.10.0