yandex-research / rtdl-num-embeddings

(NeurIPS 2022) On Embeddings for Numerical Features in Tabular Deep Learning
https://arxiv.org/abs/2203.05556
MIT License
312 stars 34 forks source link

Perform a simple prediction from the tuned and trained best model ? #6

Closed manmeet3591 closed 2 years ago

manmeet3591 commented 2 years ago

How can I perform only prediction over a test data and save the output in a csv file ? Sorry if this is too basic a question.

Yura52 commented 2 years ago

only prediction over a test data

In this line, change ['train', 'val', 'test'] to ['test']

save the output in a csv file

Could you clarify what you mean by csv in this context? The prediction variable in the line mentioned above is just a dictionary dict[str, numpy.ndarray], so you can take access the numpy array like prediction['test'].

manmeet3591 commented 2 years ago

Thank you. Also, the following line

lib.dump_predictions(predictions, output)

seems to save the outputs. What is the path where this goes ?

Yura52 commented 2 years ago

The path is output / 'predictions.npz' (see this for details). You can print output to see the content of this variable.

manmeet3591 commented 2 years ago

Thank you