Skip to content
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.

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

[BUG] download phase of movielens example fails on cpu-only system #164

Closed
mattf opened this issue Mar 26, 2022 · 0 comments · Fixed by #166
Closed

[BUG] download phase of movielens example fails on cpu-only system #164

mattf opened this issue Mar 26, 2022 · 0 comments · Fixed by #166

Comments

@mattf
Copy link

mattf commented Mar 26, 2022

https://github.com/NVIDIA-Merlin/Merlin/blob/9802eaef0515729146b7b50fd1e737b745d44525/examples/getting-started-movielens/01-Download-Convert.ipynb

ratings = df_lib.read_csv(os.path.join(INPUT_DATA_DIR, "ml-25m", "ratings.csv"))
---
ratings = ratings.drop("timestamp", axis=1)
# convert ratings to pandas df to use sklearn train_test_split func
ratings = ratings.to_pandas()
train, valid = train_test_split(ratings, test_size=0.2, random_state=42)

on a cpu-only system, ratings is already a pandas dataframe and ratings.to_pandas doesn't exist.

@mattf mattf changed the title [BUG] [BUG] download phase of movielens example fails on cpu-only system Mar 26, 2022
@karlhigley karlhigley linked a pull request Mar 26, 2022 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant