コード例 #1
0
ファイル: main.go プロジェクト: adamdrake/decisiontrees
func getAnnotation(node *pb.TreeNode) string {
	if node.GetAnnotation() == nil || node.GetAnnotation().LeftFraction == nil {
		return ""
	}

	if node.GetAnnotation().GetLeftFraction() > 0.5+*likelyThreshold {
		return "LIKELY"
	}

	if node.GetAnnotation().GetLeftFraction() < 0.5-*likelyThreshold {
		return "UNLIKELY"
	}

	return ""
}