-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.py
More file actions
24 lines (17 loc) · 826 Bytes
/
Copy pathconfig.py
File metadata and controls
24 lines (17 loc) · 826 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from os import makedirs
from os.path import join
# ---- Base Config ----
data_path = join('data', 'ml-1m', 'ratings.dat')
processed_files_dir = 'processed_data'
all_instances_file_name = 'ratings_all'
train_file_name = 'ratings_train'
test_file_name = 'ratings_test'
test_perc = 0.25 # percentage of testing instances
collaborative_neighbours = 150 # number of neighbours used in collaborative filtering
concepts = 40 # number of concepts/eigen values to consider while performing SVD decomposition
CUR_no_cols = 4 * concepts # number of columns and rows to select while performing CUR
# ----------------
txt_dir = join(processed_files_dir, 'txts')
sparse_dir = join(processed_files_dir, 'sparse_matrices')
makedirs(txt_dir, exist_ok=True)
makedirs(sparse_dir, exist_ok=True)