Closed hengzhe-zhang closed 3 years ago
UPD
There are several approaches to training and prediction:
Do you mean that I only need to wrap the FTTransformer
using skorch
?
NOTE: I have updated the previous answer, please, read it first.
Do you mean that I only need to wrap the
FTTransformer
usingskorch
?
In theory, yes. Note that rtdl.FTTransformer
expects two arguments (numerical and categorical features), so you will need to read this section.
Feel free to reopen the issue if you have more questions on the topic.
@Yura52 I implement a scikit-learn compatible interface for algorithms in this library and already open sourced it on GitHub. (https://github.com/zhenlingcn/scikit-rtdl)
This project is interesting and I want to use it as the baseline algorithm for my paper. However, it seems that I need to take several steps in order to make a prediction. Consequently, is it possible to provide a scikit-learn interface for making a convenient comparison between different algorithms?