recbole

recboleの使い方

タグ

参考

  • 様々なdeep learningベースのレコメンドをライブラリ化したパッケージ
  • kaggleのオープンカーネルでよく使われる

インストール

$ python3 -m pip install recbole

試しに動かす

ファイルを作成

from recbole.quick_start import run_recbole

run_recbole(model='BPR', dataset='ml-100k')

実行

$ python3 run.py

設定ファイル

  • yaml形式で記述する

設定ファイル例

# Training and evaluation config
epochs: 500
train_batch_size: 4096
eval_batch_size: 4096
neg_sampling:
    uniform: 1
eval_args:
    group_by: user
    order: RO
    split: {'RS': [0.8,0.1,0.1]}
    mode: full
metrics: ['Recall', 'MRR', 'NDCG', 'Hit', 'Precision']
topk: 10
valid_metric: MRR@10
metric_decimal_place: 4

設定ファイルを用いた学習用コード

from recbole.quick_start import run_recbole

run_recbole(model='BPR', dataset='ml-100k', config_file_list=['test.yaml'])

参考