-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathconfig.py
22 lines (22 loc) · 806 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
class Config:
tfidf_pipeline = 'models/ml/preprocess.sav'
model = 'models/ml/LogisticRegression.sav'
id_to_category = {0: 'course',
1: 'football',
2: 'basketball',
3: 'athletisme',
4: 'course-hippique',
5: 'equitation',
6: 'voile',
7: 'golf',
8: 'handball',
9: 'jeux-olympiques',
10: 'omnisport',
11: 'hiver',
12: 'tennis',
13: 'cyclisme',
14: 'boxe',
15: 'escrime',
16: 'judo',
17: 'volleyball',
}