from examples.base_process import main main( model_name='my_mlp', dataset_name='ML1m', dataset_path= 'E:\liudong\\feature_cross\借鉴的代码\pytorch-fm-master\data\\ml-1m\\ratings.dat', epoches=2, batch_size=2048, lr=0.001, device='cpu', save_path=None, model_params={ 'embed_size': 16, 'hidden_nbs': [400, 400, 400], 'dropout': 0 })
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, })
from examples.base_process import main main( model_name='lr', dataset_name='ML1m', dataset_path= 'E:\liudong\\feature_cross\借鉴的代码\pytorch-fm-master\data\\ml-1m\\ratings.dat', epoches=2, batch_size=2048, lr=0.001, device='cpu', save_path=None)