Exemple #1
0
def _compute_partial_predictions(
        feature_data: "np.ndarray[Any, np.dtype[np.float32]]",
        forest: VigraRandomForest) -> "np.ndarray[Any, np.dtype[np.float32]]":
    return forest.predictProbabilities(feature_data) * forest.treeCount()
Exemple #2
0
 def do_predict(forest: VigraRandomForest):
     return forest.predictProbabilities(
         feature_data.linear_raw()) * forest.treeCount()