Ejemplos de main en Python

Ejemplo n.º 1

Archivo: Train_MyMLP.py Proyecto: stormDL/rs-with-pytorch

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
    })

Ejemplo n.º 2

Archivo: Train_FM.py Proyecto: stormDL/rs-with-pytorch

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,
    })

Ejemplo n.º 3

Archivo: Train_LR.py Proyecto: stormDL/rs-with-pytorch

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)