/
run_lgbm_private.py
44 lines (41 loc) · 1.25 KB
/
run_lgbm_private.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
44
from KFT.benchmarks.lgbm import lgbm
def run():
GPU = False
params= dict()
params['gpu'] = GPU
params['its'] = 1000
params['hyperopts'] = 20
params['regression'] = True
params['num_threads'] = 0
SAVE_PATH = './lgbm_movielens'
y_name = 'rating'
data_path = 'movielens_parquet'
for i in [1,2,3,4,5]:
l = lgbm(seed=i,y_name=y_name,data_path=data_path,save_path=SAVE_PATH,params=params)
l.run()
params= dict()
params['gpu'] = GPU
params['its'] = 1000
params['hyperopts'] = 20
params['regression'] = True
params['num_threads'] = 0
SAVE_PATH = './lgbm_alcohol'
y_name = 'Bottles Sold'
data_path = 'alcohol_sales_parquet'
for i in [1,2,3,4,5]:
l = lgbm(seed=i,y_name=y_name,data_path=data_path,save_path=SAVE_PATH,params=params)
l.run()
params= dict()
params['gpu'] = GPU
params['its'] = 1000
params['hyperopts'] = 20
params['regression'] = True
params['num_threads'] = 0
SAVE_PATH = './private_lgbm'
y_name = 'total_sales'
data_path = 'benchmark_data_lgbm'
for i in [1,2,3,4,5]:
l = lgbm(seed=i,y_name=y_name,data_path=data_path,save_path=SAVE_PATH,params=params)
l.run()
if __name__ == '__main__':
run()