zhaojinghao
/
ai_platform_regression
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

修改response

This commit is contained in:
zhaojinghao 2022-12-12 10:37:51 +08:00
parent 859a3e2688
commit 998239de5f
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
run.py
View File

@ -55,11 +55,15 @@ def predict_price():
train_data = None train_data = None
else: else:
train_data = pd.read_csv(train_data) train_data = pd.read_csv(train_data)
if test_data is None or pd.read_csv(test_data).shape[0] == 0: if test_data is None:
resp_info["msg"] = "测试数据为空" resp_info["msg"] = "测试数据为空"
resp_info["code"] = 406 resp_info["code"] = 406
else: else:
test_data = pd.read_csv(test_data) test_data = pd.read_csv(test_data)
if test_data.shape[0] == 0:
resp_info["msg"] = "测试数据为空"
resp_info["code"] = 406
else:
try: try:
if train_data is None: if train_data is None:
rst = run_boston_price(test_data, None, num_boost_round, early_stopping_rounds, **params) rst = run_boston_price(test_data, None, num_boost_round, early_stopping_rounds, **params)
@ -119,7 +123,7 @@ def predict_height():
resp_info["code"] = 406 resp_info["code"] = 406
else: else:
resp_info["code"] = 200 resp_info["code"] = 200
resp_info["data"] = rst resp_info["data"] = str(rst)
resp_info["dtype"] = TEXT resp_info["dtype"] = TEXT
resp = make_response(json.dumps(resp_info)) resp = make_response(json.dumps(resp_info))
resp.status_code = 200 resp.status_code = 200
@ -159,4 +163,4 @@ def run_ts_predict():
if __name__ == '__main__': if __name__ == '__main__':
app.run(host='0.0.0.0', port=8901, debug=False) app.run(host='0.0.0.0', port=8908, debug=True)