pytorch-fm

pytorch-fm copied to clipboard

I execute python main.py under the examples/ folder, and get the error:

% python main.py --dataset_path criteo/train.txt
Traceback (most recent call last):
  File "main.py", line 189, in <module>
    main(args.dataset_name,
  File "main.py", line 151, in main
    dataset = get_dataset(dataset_name, dataset_path)
  File "main.py", line 33, in get_dataset
    return CriteoDataset(path)
  File "/Users/erlebach/opt/anaconda3/envs/torch/lib/python3.8/site-packages/torchfm/dataset/criteo.py", line 40, in __init__
    self.__build_cache(dataset_path, cache_path)
  File "/Users/erlebach/opt/anaconda3/envs/torch/lib/python3.8/site-packages/torchfm/dataset/criteo.py", line 56, in __build_cache
    feat_mapper, defaults = self.__get_feat_mapper(path)
  File "/Users/erlebach/opt/anaconda3/envs/torch/lib/python3.8/site-packages/torchfm/dataset/criteo.py", line 70, in __get_feat_mapper
    with open(path) as f:
FileNotFoundError: [Errno 2] No such file or directory: 'criteo/train.txt'

Could you please help me understand how to make this example work? Would it be possible for you to provide a single example that works out of the box, or provide an example of command line arguments that will work? Thanks.