-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
43 lines (38 loc) · 921 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import os
# Define available genres for training
GENRES = {
16: 'Animation',
35: 'Comedy',
10751: 'Family',
12: 'Adventure',
14: 'Fantasy',
10749: 'Romance',
18: 'Drama',
28: 'Action',
80: 'Crime',
53: 'Thriller',
27: 'Horror',
36: 'History',
878: 'Science Fiction',
9648: 'Mystery',
10752: 'War',
10769: 'Foreign',
10402: 'Music',
99: 'Documentary',
37: 'Western',
10770: 'TV Movie',
}
HYPERPARAMS = {
'num_units': 310,
'lr': 0.004,
'epochs': 200,
'batch_size': 768,
'threshold': 0.3
}
ENCODER_PATH = os.path.join('model','encoder')
RAW_DATASET_PATH = os.path.join('data', 'the-movies-dataset', 'movies_metadata.csv')
DATASET_PATH = os.path.join('data', 'training')
FEATURES_PATH = os.path.join('data', 'training', 'vectors')
TRAINED_MODEL_PATH = os.path.join('model', 'trained_model')
RANDOM_SEED = 42
PORT = 8081