Classification returns prediction file with two columns instead of one
Zero2848 opened this issue · comments
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.
Ok That makes sense. Thank You