from examples.base_process import main


# main(model_name='fm', dataset_name='ML20m',
#      dataset_path='C:\\Users\\dongliu\\Downloads\\ml-25m\\ml-25m\\ratings.csv',
#      epoches=100, batch_size=2048,
#      lr=0.001, device='cpu', save_path=None,
#      model_params={
#         'embed_size':16,
#     })


main(model_name='fm', dataset_name='ML1m',
     dataset_path='E:\liudong\\feature_cross\借鉴的代码\pytorch-fm-master\data\\ml-1m\\ratings.dat',
     epoches=100, batch_size=2048,
     lr=0.001, device='cpu', save_path=None,
     model_params={
        'embed_size':16,
    })