Skip to content

How to set Model Hyper-Parameters in run_recbole()? #758

Answered by hyp1231
johnny12150 asked this question in Q&A
Discussion options

You must be logged in to vote

If you use run_recbole, then you can:

  1. Write down your customized hyper-parameters into a yaml file, for example, named as your/path/to.yaml. Then
    from recbole.quick_start import run_recbole
    
    run_recbole(model='BPR', dataset='ml-100k', config_file_list=['your/path/to.yaml'])
  2. OR, pass a dict of your customized hyper-parameters directly into run_recbole, for example,
    from recbole.quick_start import run_recbole
    
    my_dict = {
      'train_batch_size': 1234
    }
    run_recbole(model='BPR', dataset='ml-100k', config_dict=my_dict)

Please refer to API Docs for more details about run_recbole.

Besides, hyper.test is designed for Parameter Tuning, but not for set models' hyper-parameters.

Replies: 1 comment 2 replies

Comment options

You must be logged in to vote
2 replies
@johnny12150
Comment options

@hyp1231
Comment options

Answer selected by johnny12150
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
good first issue Good for newcomers
2 participants