YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-12-13 00:16:44 +00:00
Code Issues Releases Wiki Activity

add files

Browse Source
This commit is contained in:
Newnius
2020-04-30 00:57:54 +08:00
parent d3d244f600
commit bd108f03fe
2 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
serve.py
View File

@@ -191,7 +191,7 @@ class MyHandler(BaseHTTPRequestHandler):
gpu_model = query.get('gpu_model')[0]
time = query.get('time')[0]
data_gainer = Data(config)
test_X, test_Y = np.array([[job, gpu_model, time],[1,1,1]]) #data_gainer.get_test_data(return_label_data=True)
test_X, test_Y = np.array([[job, gpu_model, time, 1,1,1]]) #data_gainer.get_test_data(return_label_data=True)
print(test_X, test_Y)
pred_result = predict(config, test_X)
print(pred_result)