database/openGauss-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update src/gausskernel/dbmind/tools/predictor/python/model.py.

Browse Source
This commit is contained in:
宋清怡
2021-04-27 11:17:22 +08:00
committed by Gitee
parent 37befceac8
commit 10c77e58f7
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/gausskernel/dbmind/tools/predictor/python/model.py
View File

@ -405,8 +405,7 @@ class ModelInfo:
self.model_logger.warning(
'The params of model is not complete, and the params are as following: {}'.format(params_dict))
return False
else:
return True
return True
@ -484,7 +483,7 @@ class RnnModel():
self.model_info.feature_length = feature_length
shutil.rmtree(self.model_info.model_path, ignore_errors=True)
shutil.rmtree(os.path.realpath(
os.path.join(settings.PATH_LOG, self.model_info.model_name), ignore_errors=True))
os.path.join(settings.PATH_LOG, self.model_info.model_name)), ignore_errors=True)
arr_startup = np.array(arr_startup, dtype=float).reshape((-1, 1))
arr_total = np.array(arr_total, dtype=float).reshape((-1, 1))
arr_row = np.array(arr_row, dtype=float).reshape((-1, 1))