-
Notifications
You must be signed in to change notification settings - Fork 26
Expand file tree
/
Copy pathvelox_config.py
More file actions
28 lines (25 loc) · 854 Bytes
/
velox_config.py
File metadata and controls
28 lines (25 loc) · 854 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
25
26
27
28
import json
matrixfact_config = {
'onlineUpdateDelayInMillis': 5000,
'batchRetrainDelayInMillis': 50000000,
'config': {
'numFeatures': 50
},
'modelType': 'edu.berkeley.veloxms.examples.MatrixFactorizationModel',
}
newsgroups_config = {
'onlineUpdateDelayInMillis': 5000,
'batchRetrainDelayInMillis': 50000000,
'config': {
'dataPath': 's3n://20newsgroups/',
},
'modelType': 'edu.berkeley.veloxms.examples.NewsgroupsModel',
}
config = {
'sparkMaster': "local[2]",
'sparkDataLocation': "/Users/crankshaw/Desktop/velox-data",
'models': {
'matrixfact': json.dumps(matrixfact_config),
'newsgroups': json.dumps(newsgroups_config)
}
}