Predict Continues Variable using Tabular Learner?

(Tom Ilsinszki) #1

What’s the best method to use to predicting a continuous variable using the Tabular Learner? I tried 2 methods (learn.get_preds and learn.predict), but I didn’t get predictions that seemed to be “good”.

If I try getting predictions for my test set I get all zero predictions:


[9.5214e-05]]), tensor([0, 0, 0, …, 0, 0, 0])]

FYI - I have a dataset where the variable that I’m trying to predict is 0.001 most of the times, but in some cases (maybe 5% of the cases) it could be greater than 100 or even greater than 1000 in some rare cases (not sure if this is relevant to my problem somehow).

Here’s my jupyter notebook:

For the validation set get_preds() seem to work (you can see all the 0.001 values in the predictions):


tensor([0.0010, 0.0010, 0.0010, …, 0.0010, 0.0010, 0.0010])]

I also tried the learn.predict method, but I’m not sure what that returns, but that does not look like useful predictions either.

Unfortunately the fastai jupiter notebooks for the lectures that are using the Tabular Learner do not seem to show how to generate predictions (they only show how to build the model).

Any help is appreciated.