Classification returns prediction file with two columns instead of one

kaz-Anova / StackNet

StackNet is a computational, scalable and analytical Meta modelling framework

MIT License

1.32k stars 344 forks source link

Classification returns prediction file with two columns instead of one #54

Closed Zero2848 closed 6 years ago

kaz-Anova commented 6 years ago

Not sure what you mean? StackNet returns probabilities in classification problems. So if your target variables has 2 classes (lets say 0 and 1), it returns 2 columns. The first columns is the probability to be zero and the second column the probability to be 1.

Zero2848 commented 6 years ago

Ok That makes sense. Thank You