serliezer's picture
Add config/real_data.yaml
460b790 verified
# Real data experiment configuration
lastfm:
dataset: "matthewfranglen/lastfm-1k"
max_users: 1000
max_items: 1000
max_edges: 50000
min_user_degree: 5
min_item_degree: 5
max_count: 100
K_values: [5, 10]
num_deletions: 50
radii: [1, 2, 3, 4]
prior:
a0: 0.3
b0: 1.0
c0: 0.3
d0: 1.0
max_iter: 300
tol: 1.0e-4
seed: 42
movielens_rating_count:
dataset: "ashraq/movielens_ratings"
mode: "rating_count"
max_users: 1000
max_items: 1000
max_edges: 50000
min_user_degree: 5
min_item_degree: 5
K_values: [5, 10]
num_deletions: 50
radii: [1, 2, 3, 4]
prior:
a0: 0.3
b0: 1.0
c0: 0.3
d0: 1.0
max_iter: 300
tol: 1.0e-4
seed: 42
movielens_binary:
dataset: "ashraq/movielens_ratings"
mode: "binary"
max_users: 1000
max_items: 1000
max_edges: 50000
min_user_degree: 5
min_item_degree: 5
K_values: [5, 10]
num_deletions: 50
radii: [1, 2, 3, 4]
prior:
a0: 0.3
b0: 1.0
c0: 0.3
d0: 1.0
max_iter: 300
tol: 1.0e-4
seed: 42