From 41454b6bc033a4b7c85eaa9227da5f70410940ce Mon Sep 17 00:00:00 2001 From: liyong Date: Fri, 31 Oct 2025 14:40:46 +0800 Subject: [PATCH] =?UTF-8?q?=E5=90=8C=E6=AD=A5=E6=9C=AC=E5=9C=B0=E6=9C=80?= =?UTF-8?q?=E6=96=B0=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- FC_ML_Baseline/FC_ML_Baseline_Predict/Model_Pred.py | 13 +++++++++++++ .../FC_ML_Baseline_Test/Train/param-linux.json | 12 ++++++------ FC_ML_Data/FC_ML_Data_Output/Data_Output_Pytorch.py | 2 +- 3 files changed, 20 insertions(+), 7 deletions(-) diff --git a/FC_ML_Baseline/FC_ML_Baseline_Predict/Model_Pred.py b/FC_ML_Baseline/FC_ML_Baseline_Predict/Model_Pred.py index e8c89f1..b066c16 100644 --- a/FC_ML_Baseline/FC_ML_Baseline_Predict/Model_Pred.py +++ b/FC_ML_Baseline/FC_ML_Baseline_Predict/Model_Pred.py @@ -3,6 +3,19 @@ import json import torch +import sys +import os + +# 获取当前脚本(Data_Load.py)所在的目录 +current_script_dir = os.path.dirname(__file__) # 结果:/home/app/model/ModelTrainingPython/FC_ML_Baseline/FC_ML_Baseline_Data_Handler + +# 从当前目录回退 2 级,得到项目根目录 ModelTrainingPython +root_path = os.path.abspath(os.path.join(current_script_dir, "..", "..")) + +# 将根目录添加到 Python 搜索路径 +sys.path.append(root_path) + + from FC_ML_Data.FC_ML_Data_Process.Data_Process_Normalization import Normalizer from FC_ML_NN_Model.Poly_Model import PolyModel from FC_ML_Tool.Serialization import parse_json_file diff --git a/FC_ML_Baseline/FC_ML_Baseline_Test/Train/param-linux.json b/FC_ML_Baseline/FC_ML_Baseline_Test/Train/param-linux.json index 9a0c7d9..e4a9abc 100644 --- a/FC_ML_Baseline/FC_ML_Baseline_Test/Train/param-linux.json +++ b/FC_ML_Baseline/FC_ML_Baseline_Test/Train/param-linux.json @@ -2,22 +2,22 @@ "files":["sample1.CSV"], "path": "/home/app/model/ModelTrainingPython/FC_ML_Baseline/FC_ML_Baseline_Test/Train/", "algorithmParam": { - "inputSize": 8, + "inputSize": 9, "outputSize": 8, "algorithm": "多项式拟合", "activateFun": "sigmod", "lossFun": "l1", "optimizeFun": "sgd", - "exportFormat": ".onnx", + "exportFormat": "bin", "trainingRatio": 80, "loadSize": 32, "studyPercent": 0.001, "stepCounts": 3, - "roundPrint": 11, - "round": 1001, - "preDisposeData": false, + "roundPrint": 10, + "round": 300, + "preDisposeData": true, "disposeMethod": "minmax", - "dataNoOrder": false + "dataNoOrder": true }, "algorithm": "基础神经网络NN" } \ No newline at end of file diff --git a/FC_ML_Data/FC_ML_Data_Output/Data_Output_Pytorch.py b/FC_ML_Data/FC_ML_Data_Output/Data_Output_Pytorch.py index 476d238..878e7dc 100644 --- a/FC_ML_Data/FC_ML_Data_Output/Data_Output_Pytorch.py +++ b/FC_ML_Data/FC_ML_Data_Output/Data_Output_Pytorch.py @@ -19,4 +19,4 @@ def export_model(model,target,file_name,name,input_tensor): if name == 'pth': return export_model_pt(model,target,file_name) else: - raise ValueError(f"不支持的导出类型") \ No newline at end of file + raise ValueError(f"不支持的导出类型") \ No newline at end of file