New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Dataset format #18

Open

Jason-WT opened this issue Jun 10, 2019 · 1 comment

Jason-WT commented Jun 10, 2019

What format does the model need to train and test? (Only the format of LETOR can be used in the model??)

KasSanderink commented May 13, 2020

Using the following seems to work for me:

TX: a pandas dataframe containing all features (except the target values and the group ids)
Ty: a pandas series containing the target values
Tqids: a pandas series (same length as Ty) cotaining the group id for each instance

Using numpy arrays instead of dataframes probably works as well

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment