UnixJunkie / orf

OCaml Random Forests
Other
8 stars 1 forks source link

class prediction with margin #10

Closed UnixJunkie closed 3 years ago

UnixJunkie commented 3 years ago

cf. a Breiman paper for the margin definition. This is a kind of confidence about the class prediction.

UnixJunkie commented 3 years ago

IIRC: probability of one class, minus maximum probability for the other class over all trees

UnixJunkie commented 3 years ago

done