You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I found dlrm_main.py supported two data set. The criteo_kaggle has smaller size.
parser.add_argument(
"--dataset_name",
type=str,
default="criteo_1t",
help="dataset for experiment, current support criteo_1tb, criteo_kaggle",
)
I downloaded the criteo_kaggle dataset from https://www.kaggle.com/datasets/mrkmakr/criteo-dataset. But it only contains two raw file - train.txt and test.txt. I am not sure how to process it for dlrm module to run. Could someone give me a hint?
The text was updated successfully, but these errors were encountered:
esharkwang
changed the title
How to run dlrm module with
How to run dlrm module with criteo_kaggle dataset?
Jul 27, 2023
I found dlrm_main.py supported two data set. The criteo_kaggle has smaller size.
parser.add_argument(
"--dataset_name",
type=str,
default="criteo_1t",
help="dataset for experiment, current support criteo_1tb, criteo_kaggle",
)
I downloaded the criteo_kaggle dataset from https://www.kaggle.com/datasets/mrkmakr/criteo-dataset. But it only contains two raw file - train.txt and test.txt. I am not sure how to process it for dlrm module to run. Could someone give me a hint?
The text was updated successfully, but these errors were encountered: