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)