ULTR-Community / ULTRA_pytorch

Unbiased Learning To Rank Algorithms (ULTRA)
https://ultr-community.github.io/ULTRA_pytorch/
Apache License 2.0
93 stars 8 forks source link

preprocess_data seems to be used by wrong class #13

Open we1559 opened 1 year ago

we1559 commented 1 year ago

https://github.com/ULTR-Community/ULTRA_pytorch/blob/1757587f369336ac5b76c48f9805e4e921597a3b/main.py#L93-L95

As the code above, I noticed that we use the train_input_feed class to preprocess all data, including train, valid and test data.

It seems that we should use valid_input_feed class to preprocess the valid data?

Also, all the input feed class dosn't implement the preprocess_data funciton, so it looks work well now.