Hello
I was playing with my own data following the tabular learner tutorial, and I found thie ‘issue’.
When I try to predict the class using :
row, clas, probs = learn.predict(df.iloc[0])
The objet clas and probs are the same. The method don’t return a class, returns two probabilities like this:
(tensor([0.2861]), tensor([0.2861]))
How can I get the class? or is something wrong in the method predict.
Regards!