Skip to content

Commit 60b4ca1

Browse files
Remove bpr solution
1 parent fc25a65 commit 60b4ca1

File tree

23 files changed

+231
-452
lines changed

23 files changed

+231
-452
lines changed

configs/baselines/method/bpr.yaml

Lines changed: 0 additions & 1 deletion
This file was deleted.

configs/baselines/method/simple.yaml

Lines changed: 0 additions & 1 deletion
This file was deleted.

configs/data_processing/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,4 @@ project_name: "MyProject"
1111
tags: ["${now:%Y-%m-%d}"]
1212
list_size: 5
1313
split_date: "2019/04/09"
14-
user_history_size: 5
14+
user_history_size: 10

configs/training/callbacks/default.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ early_stopping:
1010
_target_: lightning.pytorch.callbacks.EarlyStopping
1111
monitor: "auroc/val"
1212
mode: "max"
13-
patience: 5
13+
patience: 10
1414

1515
# rich_model_summary:
1616
# _target_: lightning.pytorch.callbacks.RichModelSummary

configs/training/config.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,5 @@ hydra:
1616
dir: .
1717

1818
project_name: "MyProject"
19+
tags: ["${now:%Y-%m-%d}"]
1920
data_processing_task_id: null

configs/training/datamodule/bpr.yaml

Lines changed: 0 additions & 5 deletions
This file was deleted.

configs/training/experiment/bpr_mf.yaml

Lines changed: 0 additions & 18 deletions
This file was deleted.

configs/training/experiment/bpr_mlp.yaml

Lines changed: 0 additions & 22 deletions
This file was deleted.

configs/training/experiment/simple_mf.yaml renamed to configs/training/experiment/simple_mf1.yaml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,16 @@ datamodule:
99
pin_memory: true
1010

1111
model:
12-
embed_size: 128
12+
embed_size: 64
13+
user_history_based: false
1314
lr: 0.0005
1415

1516
trainer:
1617
accelerator: "gpu"
1718
devices: 1
18-
max_epochs: 10
19+
max_epochs: 50
20+
21+
callbacks:
22+
early_stopping:
23+
patience: 10
24+
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# @package _global_
2+
3+
defaults:
4+
- override /datamodule: simple.yaml
5+
- override /model: simple_mf.yaml
6+
7+
datamodule:
8+
num_workers: 7
9+
pin_memory: true
10+
11+
model:
12+
embed_size: 64
13+
user_history_based: true
14+
lr: 0.005
15+
16+
trainer:
17+
accelerator: "gpu"
18+
devices: 1
19+
max_epochs: 50
20+
21+
callbacks:
22+
early_stopping:
23+
patience: 10
24+

0 commit comments

Comments
 (0)