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'])