From a77c498b78bd4b3063afb8186ae6df809f4dbc5e Mon Sep 17 00:00:00 2001 From: YoungHector <2581131950@qq.com> Date: Sun, 27 Oct 2024 12:00:14 +0800 Subject: [PATCH] =?UTF-8?q?=E5=90=8E=E5=A4=84=E7=90=86=E6=97=B6=E9=80=90?= =?UTF-8?q?=E6=AD=A5=E7=B2=BE=E7=AE=80=E7=89=B9=E5=BE=81=EF=BC=8C=E8=8A=82?= =?UTF-8?q?=E7=BA=A6=E5=86=85=E5=AD=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/_data_lib.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/src/_data_lib.py b/src/_data_lib.py index e9a86c9..452c412 100644 --- a/src/_data_lib.py +++ b/src/_data_lib.py @@ -469,7 +469,11 @@ def _set_model_predicted_result_pool(self, X_list, predicted_result): messager.send_to_bot_shortcut('预测结果与输入完成拼接 内存：{}'.format(get_memory_usage())) X_list.columns = ['datasource', '1', '2', '3', '4', 'year', 'month', 'day', 'hour', 'minute', 'second', '11', 'predicted_y'] - + + # 去掉所有无关数据 + X_list = X_list[['datasource', 'year', 'month', 'day', 'hour', 'minute', 'second', 'predicted_y']] + + X_list['datetime'] = pd.to_datetime(X_list[['year', 'month', 'day', 'hour', 'minute', 'second']]) messager.send_to_bot_shortcut('完成时间戳转换') messager.send_to_bot_shortcut('完成时间戳转换 内存：{}'.format(get_memory_usage())) @@ -478,10 +482,18 @@ def _set_model_predicted_result_pool(self, X_list, predicted_result): X_list['datetime_str'] = X_list['datetime'].dt.strftime('%Y-%m-%d %H:%M:%S') messager.send_to_bot_shortcut('完成时间戳字符串化') + messager.send_to_bot_shortcut('完成时间戳字符串化 内存：{}'.format(get_memory_usage())) + + # 再去掉所有无关数据 + X_list = X_list[['datasource', 'datetime_str', 'predicted_y']] + + messager.send_to_bot_shortcut('把无关列精简掉 内存：{}'.format(get_memory_usage())) X_list = X_list[X_list['datasource'] < 33].reset_index(drop=True) messager.send_to_bot_shortcut('完成datasource筛选') + + # debug print('未来一天的预测结果')